根据列名条件重命名数据框架中的大量列



我将一个XLM文件扁平化为一个数据框架,其中一个属性是包含超过50个值的列表,生成的数据框架为名为1到50的列表创建列最终数据帧的列为A、B、C、D、0、1、2、3、4、5、6、7......50岁,我想把它重命名为A、B、C、D、Role0、Role1、Role2、Role3、Role4、......Role50,基于一些逻辑

我可以像下面这样重命名编号的列,但是基于列表中的内容,这个列计数可能会变化,因此我想检查如何根据某些条件更改列名

df.rename(columns={0:'Role0', 1:'Role1', 2:'Role2', ......}, inplace=True)

也许有更好的解决方案,但是您可以这样做:
首先,声明您的dictionary:

my_dict = {0: 'Role0'}

f-string

for i in range(1, 51):
my_dict[i] = f'Role{i}'

最后做你所做的:

df.rename(columns=my_dict, inplace=True)