保存带样式的Pandas DataFrame为图片时触发ValueError错误

问题原因 #

1. 透明度值非法：当单元格值为负数时，val/350得到负数，min(1, val/350)返回负数，而RGBA透明度必须在0-1区间内。
2. 颜色格式不兼容：使用matplotlib转换表格时，matplotlib要求RGBA的RGB值为0-1的小数，你传入的rgba(255, 50, 50, ...)用了0-255的整数，导致解析失败。

修正后的代码 #

import dataframe_image as dfi
import pandas as pd

def color_negative_red(val):
    # 计算非负透明度，限制在0-1之间
    alpha = min(1, abs(val) / 350)
    # 转换RGB为0-1小数，构造matplotlib兼容的RGBA样式
    r, g, b = 1, 50/255, 50/255
    return f'background-color: rgba({r}, {g}, {b}, {alpha})'

# 精准选中MultiIndex的"Active (bps)"列
styled_df = df.style.applymap(color_negative_red, subset=pd.IndexSlice[:, 'Active (bps)'])

dfi.export(styled_df, 'test.png', table_conversion='matplotlib')

补充优化 #

如果你的需求是仅给负数单元格添加红色背景（贴合函数名color_negative_red的语义），可以调整函数逻辑：

def color_negative_red(val):
    if val < 0:
        alpha = min(1, abs(val) / 350)
        r, g, b = 1, 50/255, 50/255
        return f'background-color: rgba({r}, {g}, {b}, {alpha})'
    # 非负数单元格不设置背景色
    return ''

备选方案（无需修改颜色格式） #

如果不想调整颜色格式，可以切换到Chrome渲染模式，它原生支持CSS标准的0-255 RGBA格式：

dfi.export(styled_df, 'test.png', table_conversion='chrome')

注：此方案要求系统已安装Chrome浏览器。

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