python – 如何编辑df.columns中的几个元素

例如,列的元素是[‘a’,’b’,2006.0,2005.0,…,1995.0]

现在,我希望将float更改为int,因此列的正确元素应为[‘a’,’b’,2006,2005,…,1995]

由于这里有很多数字,我不认为重命名(columns = {‘old name’：’new name’})是个好主意.谁能告诉我如何编辑它？

解决方法:

你可以这样做：

In [49]: df
Out[49]:
   a  b  2006.0  2005.0
0  1  1       1       1
1  2  2       2       2

In [50]: df.columns.tolist()
Out[50]: ['a', 'b', 2006.0, 2005.0]

In [51]: df.rename(columns=lambda x: int(x) if type(x) == float else x)
Out[51]:
   a  b  2006  2005
0  1  1     1     1
1  2  2     2     2