使用for循环重命名数据框中的变量

import pandas as pd

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

# 获取数据框的列名列表
column_names = df.columns.tolist()

# 创建一个新的列名列表
new_column_names = []

# 使用for循环重命名列名
for column_name in column_names:
    if column_name == 'A':
        new_column_name = 'New_A'
    elif column_name == 'B':
        new_column_name = 'New_B'
    else:
        new_column_name = column_name  # 不需要重命名的列名保持不变
    new_column_names.append(new_column_name)

# 将新的列名列表赋值给数据框的列名
df.columns = new_column_names

# 打印重命名后的数据框
print(df)