DataFrame修改列名

如何使用dataframe改列名

更新：2023-06-02 14:23

一、使用rename方法

dataframe的rename方法可以修改指定列名称。其参数可以是一个字典，字典的key为旧列名，value为新列名。代码如下：

import pandas as pd

data = pd.read_csv('data.csv')
data = data.rename(columns={'old_name': 'new_name'})

其中，old_name为需要被修改的列名，new_name是新列名。

除了传入字典外，rename方法还允许直接传入函数，例如：

data = data.rename(str.upper, axis='columns')

以上代码将所有的列名都转为大写字母。

二、使用columns属性

dataframe的columns属性是一个Index类型对象，可以直接修改其中的元素来更改列名。示例如下：

data = pd.read_csv('data.csv')
data.columns = ['new_name1', 'new_name2', 'new_name3']

以上代码将原列名依次改为新列名。

需要注意的是，使用该方法修改列名时，要确保新列名的长度与原列名相同。

三、使用set_axis方法

dataframe的set_axis方法是用于修改行或列索引的方法，也可以用来修改列名。其参数为一个列表或数组。

data = pd.read_csv('data.csv')
new_columns = ['new_name1', 'new_name2', 'new_name3']
data = data.set_axis(new_columns, axis='columns')

四、使用add_prefix和add_suffix方法

dataframe的add_prefix和add_suffix方法可以为原有列名添加前缀或后缀。示例如下：

data = pd.read_csv('data.csv')
data = data.add_suffix('_new')

以上代码会在所有列名后添加"_new"的后缀。

五、使用rename_axis方法

dataframe的rename_axis方法可以修改列索引或行索引的名称，进而影响列名的显示。示例如下：

data = pd.read_csv('data.csv')
data = data.rename_axis('new_axis_name', axis='columns')

以上代码修改了列索引的名称为"new_axis_name"。

六、总结

以上就是使用dataframe修改列名的几种方法。我们可以根据具体的场景选择不同的方法来完成任务。