Pandas修改DataFrame列名为指定范围的数
最近在用python写一些关于数据处理的东西,对pandas用得比较多。在使用过程中,经常需要修改df的列名以方便索引。经过搜索后我发现,网上的博文修改列名的用例都是可枚举的,也就是只有几个key-value对。很显然,当DataFrame的列很多的时候,一一去枚举就显得不显示了。如果需要将列名修改为指定范围的数,可以按照如下方法:
oldColumns=df.columns.tolist()#获取原来的columns并转换成列表
newColumns=[x for x in range(0,df.shape[1])]#使用列表推导式生成指定范围的数用作新的columns
map=dict(zip(oldColumns,newColumns))#这里很关键,使用该步骤将上面2个列表构造成map
df1.rename(columns=map,inplace=True)#直接使用rename方法,给columns赋值为上面得到的map
上面的注释已经解释得比较详细了,这里再强调下列表推导式里面range()函数的第二个参数肯定要和df的列数相同,所以range()函数的第二个参数是df.shape[1].
另外需要注意的是最后一行的inplace参数,设置成True是替换原来的df,如果不想替换的话可以设置成False,请根据实际情况进行修改。