字典传递映射替换列名

51 阅读1分钟

你可以使用 pandas.DataFrame.rename() 方法,通过字典传递映射关系来替换列名。例如:

import pandas as pd

# 创建示例 DataFrame
df = pd.DataFrame({
    'category': ['A', 'B', 'C'],
    'categoryCode': [101, 102, 103],
    'price': [10, 20, 30]
})

# 定义列名映射字典
column_mapping = {
    'category': '品类',
    'categoryCode': '品类代码'
}

# 使用 rename 方法替换列名
df.rename(columns=column_mapping, inplace=True)

print(df)

运行结果:

   品类  品类代码  price
0   A   101     10
1   B   102     20
2   C   103     30

这样 category 被替换为 品类categoryCode 被替换为 品类代码,未在字典中的列名(如 price)保持不变。