如何使用Pandas to_excel为单元格条件设置背景色与字体颜色？

原始DataFrame #

import numpy as np 
import pandas as pd

df = pd.DataFrame(
    {
        'a': [2, 2, 2, -4, np.nan, np.nan, 4, -3, 2, -2, -6],
        'b': [2, 2, 2, 4, 4, 4, 4, 3, 2, 2, 6]
    }
)

需求与现有实现 #

需求：仅对a列单元格设置样式：

• 当df.a > 0时，背景色为绿色
• 当df.a < 0时，背景色为红色
• 同时需要添加字体颜色设置，但不想重复编写条件逻辑，寻求更优实现

现有仅设置背景色的代码：

def highlight_cells(s):
    if s.name=='a':
        conds = [s > 0, s < 0, s == 0]
        labels = ['background-color: lime', 'background-color: pink', 'background-color: gold']
        array = np.select(conds, labels, default='')
        return array
    else:
        return ['']*s.shape[0]

df.style.apply(highlight_cells).to_excel('df.xlsx', index=False)

优化实现方案 #

更优的核心思路是将背景色和字体颜色的样式规则与对应条件绑定在一起，避免重复判断条件，让代码更简洁易维护

优化后的样式函数 #

def highlight_a_column(s):
    # 非a列直接返回空样式
    if s.name != 'a':
        return [''] * len(s)
    
    # 定义每个条件对应的完整样式（背景色 + 字体颜色）
    condition_styles = [
        (s > 0, 'background-color: lime; color: white'),    # 正数：绿背景+白色字体
        (s < 0, 'background-color: pink; color: black'),     # 负数：粉背景+黑色字体
        (s == 0, 'background-color: gold; color: darkblue')  # 零：金背景+深蓝色字体
    ]
    
    # 拆分条件和样式，传入np.select
    conditions, styles = zip(*condition_styles)
    return np.select(conditions, styles, default='')  # NaN用默认空样式

使用方式 #

df.style.apply(highlight_a_column).to_excel('df_styled.xlsx', index=False)

方案优势 #

1. 避免重复逻辑：每个条件只定义一次，同时对应背景和字体样式，无需多次判断s>0/s<0等条件
2. 易维护：要修改样式或添加新条件，只需在condition_styles列表中新增/修改条目即可
3. 可读性强：条件和对应样式一一对应，逻辑清晰直观

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