基于布尔列高亮DataFrame行的代码报错问题排查

嘿，我来帮你搞定这个高亮的问题！你遇到的报错核心原因是样式返回的形状和DataFrame不匹配——你的函数只返回了单个字符串，但apply(axis=1)要求每行返回和列数一样多的样式值（每个单元格对应一个样式），所以才会出现(281556,) vs (281556, 6)的形状错误。

修正后的代码方案 #

我们只需要修改函数，让它返回对应每行所有列的样式列表就行，这里有两种简单的实现方式：

方式1：修改原有函数，返回整行样式列表

def highlight_late(row):
    # 根据延迟状态确定颜色
    bg_color = 'background-color: red' if row['Late Submission'] else 'background-color: green'
    # 为当前行的每一列都返回这个样式
    return [bg_color] * len(row)

# 应用样式到DataFrame
styled_df = df7.style.apply(highlight_late, axis=1)
# 在Jupyter环境中直接运行styled_df即可查看效果
styled_df

方式2：用numpy简化样式矩阵生成（更高效）

如果你的DataFrame行数很多（比如你这里的28万行），用numpy广播生成样式矩阵会更高效：

import numpy as np

def highlight_late(df):
    # 生成和DataFrame形状一致的样式矩阵
    # 利用[:, None]把一维的布尔列转成二维，方便广播到所有列
    style_matrix = np.where(
        df['Late Submission'].values[:, None],
        'background-color: red',
        'background-color: green'
    )
    return style_matrix

# axis=None表示把整个DataFrame作为输入处理
styled_df = df7.style.apply(highlight_late, axis=None)
styled_df

为什么原来的代码不行？ #

你之前的函数只返回了一个'background-color: red'或green字符串，相当于每行只给了1个样式值，但你的DataFrame有6列，pandas期望每行返回6个样式值（对应每个单元格），所以就出现了形状不匹配的报错。

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