import pandas as pd
def replace_column_names(df, replace_dict):
"""
替换 DataFrame 列名中的指定字段。
:param df: pandas DataFrame
:param replace_dict: 需要替换的字段及其替换值的字典
:return: 替换后的 DataFrame
"""
df = df.rename(columns=lambda col: replace_column(col, replace_dict))
return df
def replace_column(col_name, replace_dict):
"""
替换列名中的指定字段。
:param col_name: 原列名
:param replace_dict: 替换字典
:return: 替换后的列名
"""
for key, value in replace_dict.items():
col_name = col_name.replace(key, value)
return col_name
# 示例用法
data = {
"price环比天猫": [100, 200],
"count同比京东": [50, 60],
"total销售额": [500, 600]
}
replace_dict = {"price": "价格", "count": "量", "total": "额"}
df = pd.DataFrame(data)
print("替换前列名:", df.columns.tolist())
df = replace_column_names(df, replace_dict)
print("替换后列名:", df.columns.tolist())
