pandas修改DataFrame列名的方法

  • Post category:Python

下面是“pandas修改DataFrame列名的方法”的完整攻略。

1. 用rename()方法

rename()方法是pandas中修改DataFrame列名的常用方法。这个方法返回重命名后的DataFrame副本,如需永久修改,需要用inplace=True参数。

下面是一个示例代码:

import pandas as pd

# 创建一个DataFrame
df = pd.DataFrame({'A':[1,2,3], 'B':[4,5,6]})

# 输出原始列名
print(df.columns)

# 用rename()方法修改列名
df.rename(columns={'A':'a', 'B':'b'}, inplace=True)

# 输出修改后的列名
print(df.columns)

以上代码中,我们首先创建了一个DataFrame对象,包含两个原始的列名AB。然后,我们使用rename()方法把列名AB分别修改为了ab,并用inplace=True参数把修改结果永久应用到原始DataFrame对象,最后输出了修改后的列名。

注意:rename()方法返回的是修改后的DataFrame副本,需要用inplace=True参数才能实现永久修改。

2. 直接修改.columns属性

还有一种修改DataFrame列名的方法是直接访问.columns属性并修改其中的元素。

下面是一个示例代码:

import pandas as pd

# 创建一个DataFrame
df = pd.DataFrame({'A':[1,2,3], 'B':[4,5,6]})

# 输出原始列名
print(df.columns)

# 直接修改.columns属性
df.columns = ['a', 'b']

# 输出修改后的列名
print(df.columns)

以上代码同样创建了一个DataFrame对象,包含两个原始的列名AB。然后,我们通过修改.columns属性把列名AB分别修改为了ab,最后输出了修改后的列名。

注意:直接修改.columns属性实现的是永久修改,不需要使用inplace=True参数。

以上就是“pandas修改DataFrame列名的方法”的完整攻略,希望对你有所帮助。