如何将Pandas单元格背景色复制到python-pptx表格中

核心思路 #

先从DataFrame中提取每个单元格的背景色RGB值，再对应到python-pptx表格的单元格设置填充色。以下分两种常见场景给出实现方案：

场景1：DataFrame背景色由Pandas Styler设置 #

如果你的DataFrame是通过df.style相关方法（如apply、background_gradient）设置的单元格背景色，按以下步骤操作：

步骤1：提取单元格背景色RGB值 #

import pandas as pd
from pptx.dml.color import RGBColor

# 示例DataFrame（替换为你的实际数据）
df = pd.DataFrame({
    'colA': ['abc', 'xyz'],
    'colB': ['123', 'a1a'],
    'colC': ['bca', 'zya']
})
# 示例样式设置（替换成你实际的样式逻辑）
styled_df = df.style.apply(lambda x: ['background-color: #ffcccc' if col in ['colA', 'colB'] else '' for col in x.index], axis=1)

# 提取每个数据单元格的RGB颜色值
cell_colors = {}
for row_idx in range(len(df)):
    row_styles = styled_df.export()[row_idx]
    for col_idx in range(len(df.columns)):
        style = row_styles[col_idx]
        bg_color = style.get('background-color')
        if bg_color and bg_color.startswith('#'):
            # 十六进制转RGB元组
            r = int(bg_color[1:3], 16)
            g = int(bg_color[3:5], 16)
            b = int(bg_color[5:7], 16)
            cell_colors[(row_idx, col_idx)] = (r, g, b)

步骤2：对应设置PPT表格颜色 #

假设你已创建好与DataFrame结构一致的PPT表格（第一行为表头，后续为数据行）：

# 处理数据行
for row_idx in range(len(df)):
    pptx_row_num = row_idx + 1  # PPT表格第一行是表头，数据从第二行开始
    for col_idx in range(len(df.columns)):
        cell = table.cell(pptx_row_num, col_idx)
        fill = cell.fill
        fill.solid()
        rgb_tuple = cell_colors.get((row_idx, col_idx))
        if rgb_tuple:
            fill.fore_color.rgb = RGBColor(*rgb_tuple)
        else:
            # 无颜色时设置默认白色（可按需调整）
            fill.fore_color.rgb = RGBColor(255, 255, 255)

# （可选）同步表头背景色
for col_idx in range(len(df.columns)):
    header_cell = table.cell(0, col_idx)
    fill = header_cell.fill
    fill.solid()
    header_style = styled_df._header_styles[col_idx]
    bg_color = header_style.get('background-color')
    if bg_color and bg_color.startswith('#'):
        r = int(bg_color[1:3], 16)
        g = int(bg_color[3:5], 16)
        b = int(bg_color[5:7], 16)
        fill.fore_color.rgb = RGBColor(r, g, b)

场景2：DataFrame来自带格式的Excel文件 #

如果DataFrame是从带单元格背景色的Excel文件读取的，用openpyxl提取颜色信息：

步骤1：从Excel提取单元格颜色 #

from openpyxl import load_workbook

# 加载Excel文件，保留格式信息
wb = load_workbook('your_excel_file.xlsx', data_only=True)
ws = wb.active

cell_colors = {}
# Excel行/列是1-based，转换为DataFrame的0-based索引
for excel_row in range(2, ws.max_row + 1):  # 跳过Excel表头行
    df_row_idx = excel_row - 2
    for excel_col in range(1, ws.max_column + 1):
        df_col_idx = excel_col - 1
        cell = ws.cell(row=excel_row, column=excel_col)
        fill = cell.fill
        # 排除默认无填充的单元格
        if fill.start_color.index != '00000000':
            # 提取ARGB中的RGB部分（去掉Alpha通道字符）
            argb = fill.start_color.index
            r = int(argb[2:4], 16)
            g = int(argb[4:6], 16)
            b = int(argb[6:8], 16)
            cell_colors[(df_row_idx, df_col_idx)] = (r, g, b)

步骤2：设置PPT表格颜色 #

直接使用场景1步骤2的代码，用上面提取的cell_colors字典对应设置即可。

内容的提问来源于stack exchange，提问作者Jlow