为数据框中的列变量着色并将其导出到Excel

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

# 创建一个示例数据框
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
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 = ['FF0000', '00FF00', '0000FF']  # 红色、绿色、蓝色

# 为每列着色
for idx, col in enumerate(df.columns, start=1):
    ws.column_dimensions[col].width = 15
    for cell in ws[f'{col}1']:  # 标题行
        cell.fill = PatternFill(start_color=colors[idx-1], end_color=colors[idx-1], fill_type="solid")

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