import pandas as pd
# 创建一个示例DataFrame
data = {
'A': [1, 2, 3],
'B': [4, 5, 6]
}
df = pd.DataFrame(data)
# 使用rename()函数修改列名
df = df.rename(columns={'A': 'X', 'B': 'Y'})
print(df)
在这个例子中,我们使用rename()函数将列名’A’修改为’X’,将列名’B’修改为’Y’。columns参数接受一个字典,其中键是原始列名,值是新列名。rename()函数返回一个新的DataFrame,所以我们需要将其赋值给原始变量df。如果你不想创建一个新的DataFrame,可以使用inplace=True参数直接在原始DataFrame上进行修改。例如:
df.rename(columns={'A': 'X', 'B': 'Y'}, inplace=True)