为数据框架的列名创建列表，同时将多个值更改为一个值

Question

我有一个名为df的数据框架，它是多个.csv文件的组合，因此对于每个文件中的特定索引，有几个列名。假设不同的。csv文件的列名是A, B, C, D, E。我想把列名称中的A, B, C, D, E全部改为f。

I tried this;

df = pd.read_csv(path + config['file'] + '.csv')

list = [c.replace('A', 'F') for c in df.columns]

但不能想出一个简单的一行方法将B, C, D, E的值转换成f

感谢您的帮助。

Answer 1

先定义颜色名称:

to_replace_cols=['A','B','C','D','E']
df.columns = ['F' if i in to_replace_cols else i for i in df.columns]
#one line
df.columns = ['F' if i in ['A','B','C','D','E'] else i for i in df.columns]