修改dataframe的列名

1. 重命名所有列
   如果你想要重命名所有列，可以直接赋值给 columns 属性：

import pandas as pd

# 创建示例DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})

# 新的列名列表
new_columns = ['X', 'Y', 'Z']

# 修改列名
df.columns = new_columns

print(df)

2. 重命名部分列
   如果你只想重命名部分列，可以使用 rename 方法，并传入一个字典，字典的键是旧列名，值是新列名：

import pandas as pd

# 创建示例DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})

# 创建重命名字典
rename_dict = {'A': 'X', 'B': 'Y'}

# 重命名部分列
df = df.rename(columns=rename_dict)

print(df)

3. 使用 set_axis 方法
   set_axis 方法可以同时重命名行和列，传入新列名列表以及 axis=1：

import pandas as pd

# 创建示例DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})

# 新的列名列表
new_columns = ['X', 'Y', 'Z']

# 修改列名
df = df.set_axis(new_columns, axis=1, inplace=False)

print(df)