Python数据处理之轻松实现颜色与数字的映射

color_to_num = {
    "经典黑": 1,
    "气质蓝": 2,
    "橙": 6,
    # 其他颜色映射...
}

import pandas as pd

excel_path = "source_data.xlsx"   # 可根据实际路径修改
df = pd.read_excel(excel_path)

def colors_to_numbers(color_str):
    colors = [c.strip() for c in color_str.split('，')]   # 按中文逗号分割
    numbers = [str(color_to_num.get(c, "?")) for c in colors]  # 未知颜色标记为 "?"
    return ','.join(numbers)

df["模拟结果"] = df["源数据"].apply(colors_to_numbers)

print("处理后的结果：")
print(df)

output_path = "result_with_mapping.xlsx"
df.to_excel(output_path, index=False)
print(f"\n已保存结果至：{output_path}")

import pandas as pd

# 1. 定义颜色到数字的映射字典
color_to_num = {
    "经典黑": 1, "气质蓝": 2, "橙": 6, "浪漫红": 3, "贵族金": 4,
    "蓝色": 2, "黑色": 1, "金色": 4, "米白色": 5, "酒红色": 3,
    "洋红": 3, "白": 5, "黑": 1, "蓝": 2, "红": 3, "雪白": 5,
    "玫瑰金": 4, "天际蓝": 2, "活力橙": 6, "黄金": 4,
    "金黄色": 4, "鱼肚白": 5, "白花花": 5, "红色的": 3, "大红色的": 3,
    "纯白色": 5, "灰白": 5
}

# 2. 读取Excel文件（请确保文件路径正确，且包含“源数据”列）
excel_path = "source_data.xlsx"   # 可以修改为实际文件路径
df = pd.read_excel(excel_path)

# 3. 将“源数据”列中的颜色字符串转换为数字串
def colors_to_numbers(color_str):
    colors = [c.strip() for c in color_str.split('，')]   # 中文逗号分割
    numbers = [str(color_to_num.get(c, "?")) for c in colors]
    return ','.join(numbers)

df["模拟结果"] = df["源数据"].apply(colors_to_numbers)

# 4. 输出最终结果
print("处理后的结果：")
print(df)

# 5. 可选：保存为新Excel文件
output_path = "result_with_mapping.xlsx"
df.to_excel(output_path, index=False)
print(f"\n已保存结果至：{output_path}")