如何編輯df.columns

Question

例如幾個元素，列的元素是['a', 'b', 2006.0, 2005.0, ... ,1995.0]

現在，我希望改變浮子爲int，所以列的正確內容應['a', 'b', 2006, 2005, ... , 1995]

由於這裏有很多數字，我不認爲rename(columns={'old name': 'new name'})是一個好主意。任何人都可以告訴我如何編輯它？

Answer 1

你可以這樣做：

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 

Answer 2

我認爲你可以使用list comprehension：

print (df.columns.tolist()) 
['a', 'b', 2006.0, 2005.0, 1995.0] 

print ([int(col) if type(col) == float else col for col in df.columns]) 
['a', 'b', 2006, 2005, 1995] 

df.columns = [int(col) if type(col) == float else col for col in df.columns] 
['a', 'b', 2006, 2005, 1995]