如何使用Pandas为Excel中隔组（偶数组）设置背景色

用户提供的DataFrame如下：

import pandas as pd 
import numpy as np 

df = pd.DataFrame(
    {
        'p': ['short', np.nan, 'short', np.nan, np.nan, 'long', 'long', np.nan, np.nan],
        's': [13, 13, 14, 15, 100, 1, 1000, 12, 1111]
    }
)

需求：基于p列进行分组，规则为**p列的非空值及其下方所有NaN值为一组**，导出到Excel时为偶数组设置不同背景色。

用户尝试的代码未生效：

from matplotlib import colors
def colr(x):
    y = x.assign(k=x['p'].ne(x['p'].shift()).cumsum())

    d = dict(enumerate(colors.cnames))
    y[:] = np.broadcast_to(y['k'].map(d).radd('background-color:').to_numpy()[:,None]
                          ,y.shape)
    return y.drop("k",1)

df = df.style.apply(colr,axis=None)
df.to_excel('file.xlsx', index=False, engine='openpyxl')

原代码核心问题是分组逻辑错误：x['p'].ne(x['p'].shift()).cumsum()会把每个NaN行判定为新组（NaN与前一个值不相等），无法实现“非空值+下方NaN”为一组的需求。

正确实现步骤：

1. 生成正确分组标识：用p.notna().cumsum()，每遇到p的非空值就创建新组，后续NaN自动归为同一组
2. 标记偶数组：通过组号对2取余判断是否为偶数组
3. 生成样式矩阵：为偶数组的所有单元格设置背景色，其余单元格留空

实现代码：

import pandas as pd
import numpy as np

df = pd.DataFrame(
    {
        'p': ['short', np.nan, 'short', np.nan, np.nan, 'long', 'long', np.nan, np.nan],
        's': [13, 13, 14, 15, 100, 1, 1000, 12, 1111]
    }
)

def highlight_even_groups(x):
    # 生成分组号：非空值触发新组
    group_ids = x['p'].notna().cumsum()
    # 判断是否为偶数组
    is_even_group = group_ids % 2 == 0
    # 创建样式矩阵：偶数组设置浅灰色背景，其余为空
    style_matrix = np.where(is_even_group[:, None], 'background-color: #f0f0f0', '')
    return style_matrix

# 应用样式并导出
styled_df = df.style.apply(highlight_even_groups, axis=None)
styled_df.to_excel('file.xlsx', index=False, engine='openpyxl')

代码说明：

• group_ids会生成[1,1,2,2,2,3,3,3,3]，完全匹配需求的分组规则
• is_even_group标记第2、4组（偶数组）为True
• 通过np.where批量生成样式矩阵，确保整组单元格统一染色

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