如何为Pandas透视表DataFrame设置自定义样式？报错修复

错误原因 #

你在set_properties的subset参数中使用了列的位置索引j，但你的DataFrame列名是'AUG-2022'这类字符串标签，而非数字索引。Styler无法通过数字索引找到对应的列，因此触发KeyError。

修复方案 #

方案1：修正循环中的列引用 #

将循环里的列位置索引替换为实际列名，修改后的代码如下：

styler = df.style
# 遍历每一行
for i, row in enumerate(df.values):
    # 从第二列开始遍历（跳过第一列，无前置列可比较）
    for j in range(1, len(row)):
        current_col_name = df.columns[j]
        if row[j] < row[j-1]:
            styler.set_properties(subset=[(i, current_col_name)], **{'background-color': 'lightcoral'})
        elif row[j] > row[j-1]:
            styler.set_properties(subset=[(i, current_col_name)], **{'background-color': 'lightgreen'})

styler

方案2：使用Styler.apply（推荐，性能更优） #

避免逐单元格循环，采用向量化方式生成样式，适合数据量较大的场景：

def compare_with_previous(row):
    # 初始化样式列表，默认无样式
    styles = [''] * len(row)
    # 从第二列开始与前一列比较
    for idx in range(1, len(row)):
        if row[idx] > row[idx-1]:
            styles[idx] = 'background-color: lightgreen'
        elif row[idx] < row[idx-1]:
            styles[idx] = 'background-color: lightcoral'
    return styles

# 按行应用样式函数
styler = df.style.apply(compare_with_previous, axis=1)
styler

或者用更简洁的向量化写法（利用diff计算差值）：

def generate_diff_styles(df):
    # 计算每行中当前列与前一列的差值
    col_diff = df.diff(axis=1)
    # 根据差值正负生成对应样式
    green_styles = col_diff.gt(0).applymap(lambda x: 'background-color: lightgreen' if x else '')
    red_styles = col_diff.lt(0).applymap(lambda x: 'background-color: lightcoral' if x else '')
    # 合并样式
    return green_styles + red_styles

styler = df.style.apply(generate_diff_styles, axis=None)
styler

内容的提问来源于stack exchange，提问作者RH-NewDeveloper