如何将熊猫保存为不同颜色的excel

import pandas as pd
from openpyxl.styles import PatternFill
from openpyxl import Workbook

# 创建一个熊猫数据框
data = {'Name': ['Tom', 'Jerry', 'Alice', 'Bob'],
        'Color': ['Red', 'Blue', 'Green', 'Yellow']}
df = pd.DataFrame(data)

# 创建一个Excel工作簿
wb = Workbook()
ws = wb.active

# 将熊猫数据框写入Excel工作表
for r in dataframe_to_rows(df, index=False, header=True):
    ws.append(r)

# 设置颜色
colors = {'Red': 'FFFF0000', 'Blue': 'FF0000FF', 'Green': 'FF00FF00', 'Yellow': 'FFFFFF00'}
for row in ws.iter_rows(min_row=2, min_col=2, max_col=2):
    for cell in row:
        color = colors.get(cell.value, 'FFFFFFFF')
        cell.fill = PatternFill(start_color=color, end_color=color, fill_type='solid')

# 保存Excel文件
wb.save('pandas_colors.xlsx')