python - 除某些列外，如何为列名添加后缀？

Question

给定pandas DataFrame，如何将后缀“_old”添加到除两列Id和之外的所有列Name？

import pandas as pd
data = [[1,'Alex',22,'single'],[2,'Bob',32,'married'],[3,'Clarke',23,'single']]
df = pd.DataFrame(data,columns=['Id','Name','Age','Status'])

标签： pythonpandas

Answer 1

您可以重新分配数据框的列：

>>> df
   Id    Name  Age   Status
0   1    Alex   22   single
1   2     Bob   32  married
2   3  Clarke   23   single
>>> 
>>> keep_same = {'Id', 'Name'}
>>> df.columns = ['{}{}'.format(c, '' if c in keep_same else '_old')
...:               for c in df.columns]
>>>
>>> df
   Id    Name  Age_old Status_old
0   1    Alex       22     single
1   2     Bob       32    married
2   3  Clarke       23     single