基于跨列条件的Pandas Styling样式设置ValueError报错问题咨询

问题原因 #

报错是因为你定义的date_pii函数仅在row['Date PI'] < datetime.now()条件成立时返回样式列表，不满足条件时无返回值，默认返回None，无法匹配Pandas要求的「每行返回与列数等长的样式数组」规则，你的Dataframe共10列，因此期望每行返回10个样式值，不符合条件的行无返回就会触发形状不匹配错误。

同时要注意样式的优先级：后应用的样式会覆盖先应用的样式，因此跨列判断的标红逻辑要放在所有单列样式设置之后，才能保证红底可以覆盖之前给Date PII设置的橙/黄底，符合需求。

解决方案 #

方案1：修正原函数 #

调整date_pii函数，确保所有分支都返回全长度的样式列表：

def date_pii(row):
    # 先初始化全空的样式列表，长度等于当前行的列数
    ret = ["" for _ in row.index]
    # 增加非空判断避免空值对比报错
    if pd.notna(row['Date PI']) and row['Date PI'] < datetime.now():
        ret[row.index.get_loc("Date PII")] = "background-color: red"
    # 所有分支都必须返回ret，不能只在if逻辑内返回
    return ret

styler = df3.style \
        .applymap(lambda x: 'background-color: %s' % 'red' if x <= datetime.now() else '', subset=['Date PI']) \
        .applymap(lambda x: 'background-color: %s' % 'yellow' if x < datetime.now() + timedelta(days=30) else '',
                  subset=['Date PII']) \
        .applymap(lambda x: 'background-color: %s' % 'orange' if x <= datetime.now() else '', subset=['Date PII']) \
        .applymap(lambda x: 'background-color: %s' % 'grey' if pd.isnull(x) else '', subset=['Date PI'])\
        .applymap(lambda x: 'background-color: %s' % 'grey' if pd.isnull(x) else '', subset=['Date PII'])\
        # 跨列条件放在最后应用，保证优先级最高
        .apply(date_pii, axis=1)

styler.to_excel(writer, sheet_name='Report Paris', index=False)

方案2：更简洁的向量化写法（性能更高） #

不需要逐行遍历，直接针对Date PII列生成样式，逻辑更清晰：

import numpy as np

def highlight_pii(s):
    # s是Date PII列的序列，直接用Date PI列做判断条件
    return np.where(
        (pd.notna(df3['Date PI'])) & (df3['Date PI'] < datetime.now()),
        'background-color: red',
        ''
    )

styler = df3.style \
        .applymap(lambda x: 'background-color: %s' % 'red' if x <= datetime.now() else '', subset=['Date PI']) \
        .applymap(lambda x: 'background-color: %s' % 'yellow' if x < datetime.now() + timedelta(days=30) else '',
                  subset=['Date PII']) \
        .applymap(lambda x: 'background-color: %s' % 'orange' if x <= datetime.now() else '', subset=['Date PII']) \
        .applymap(lambda x: 'background-color: %s' % 'grey' if pd.isnull(x) else '', subset=['Date PI'])\
        .applymap(lambda x: 'background-color: %s' % 'grey' if pd.isnull(x) else '', subset=['Date PII'])\
        # 直接指定subset为Date PII列，不需要整行处理
        .apply(highlight_pii, subset=['Date PII'])

styler.to_excel(writer, sheet_name='Report Paris', index=False)

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