如何修复DataFrame渐变黄色样式无法导出至Excel的问题？

核心问题 #

1. 样式覆盖顺序错误：先应用渐变黄色样式再应用灰色分级样式，导致灰色样式覆盖了0-1区间的渐变黄色效果。
2. 无效CSS生成：gradient_color函数中，数值不在0-1范围时返回的rgb({r},{g},{b})包含None，这会生成OpenPyXL无法解析的无效背景色声明。
3. 渐变方向搞反：当前RGB计算逻辑是低值（0）返回白色、高值（1）返回黄色，和需求的“低值黄色到高值白色”相反。

修复后的代码 #

def highlight_negative(val):
    if val > 1:
        if val < 500:
            color = 'dimgrey'
        elif 500 <= val < 1000: 
            color = 'lightgray'
        else:
            color = 'white'  # 用white替代None，避免无效样式声明
    else:
        color = None  # 0-1区间不设置灰色，留给渐变样式
    
    return f'background-color: {color}' if color else ''


def gradient_color(val):
    # 实现低值黄色到高值白色的渐变
    if 0 <= val <= 1:
        min_val = 0
        max_val = 1
        range_val = max_val - min_val
        if range_val == 0:
            # 所有值相等时返回白色
            return 'background-color: rgb(255,255,255)'
        else:
            # 黄色(rgb(255,255,0))到白色(rgb(255,255,255))，调整b通道从0到255
            r = 255
            g = 255
            b = int(255 * (val - min_val) / range_val)
        return f'background-color: rgb({r},{g},{b})'
    else:
        # 非0-1区间返回空，不覆盖灰色样式
        return ''

# 调整样式应用顺序：先设置灰色分级，再覆盖渐变黄色
styled_df = df.style.applymap(highlight_negative, subset=df.columns[2:])\
                    .applymap(gradient_color, subset=df.columns[2:])\
                    .format(precision=2)

styled_df.to_excel('styled_df.xlsx', engine='openpyxl')

关键修改说明 #

• 调整样式顺序：先调用highlight_negative设置>1数值的灰色背景，再调用gradient_color覆盖0-1区间的背景色，确保渐变样式能正常显示。
• 修复无效CSS：数值不在目标区间时返回空字符串而非带None的RGB值，避免OpenPyXL解析失败。
• 修正渐变方向：将b通道计算改为(val - min_val)/range_val，让val=0时b=0（黄色）、val=1时b=255（白色），符合需求。
• 替换无效值：highlight_negative中val>=1000时用white替代None，避免生成无效的样式声明。

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