python - 重命名 Pandas 数据框中的选定列

Question

假设我有以下数据框:

import pandas as pd
raw_data = {'col1': [23,45,21,45,23],'col2': [3,4,5,22,3],'col3': ['y1', 'y2', 'y3', 'y1', 'y2'],'col4': ['x1', 'x2', 'x3', 'x1', 'x2']}
df = pd.DataFrame(raw_data, columns = ['col1','col2','col3','col4','col5'])

现在我想以编程方式为除 'col2' 和 'col4' 之外的每个列名称添加前缀 'PRE-'。

是否有任何方法/功能可以实现这一点？

感谢您的指点

Answer 1

使用 DataFrame.rename带 lambda 函数:

df = df.rename(columns = lambda x: f'PRE-{x}' if x not in ['col2','col4'] else x)
print (df)
   PRE-col1  col2 PRE-col3 col4 PRE-col5
0        23     3       y1   x1      NaN
1        45     4       y2   x2      NaN
2        21     5       y3   x3      NaN
3        45    22       y1   x1      NaN
4        23     3       y2   x2      NaN