Pandas DataFrame按条件设行样式导出Excel异常求助

问题现象 #

根据条件为Pandas DataFrame设置行样式并导出至Excel后，Excel文件中显示的是False和background-color:yellow字符串，而非带有黄色背景的原始数据。

错误原因 #

• 原代码中highlight_cells函数错误地基于存储样式标记（布尔值/字符串）的DataFrame创建Styler对象，而非基于原始数据的DataFrame来应用样式规则。
• 最终导出的是样式标记内容，而非应用了样式的原始数据。

修正方案 #

核心是基于原始数据生成Styler对象，再通过行级样式函数按条件应用背景色：

修正后的完整代码 #

def highlight():
    df1, df2 = read_df()
    filtered_df = filter(df1, df2)
    # 基于原始数据df1创建样式对象，传入过滤结果作为条件
    styled_df = df1.style.apply(highlight_rows, filtered=filtered_df, axis=1)
    
    with pd.ExcelWriter(r"C:\Users\test.xlsx", engine="openpyxl", mode="a", if_sheet_exists="replace") as writer:
        styled_df.to_excel(writer, sheet_name='result', index=False)


def highlight_rows(row, filtered):
    yellow_style = 'background-color: yellow'
    # 判断当前行是否属于过滤后的结果（整行匹配）
    # 如需匹配任意列存在交集，替换为: if row.isin(filtered.values.flatten()).any()
    if row.name in filtered.index and row.equals(filtered.loc[row.name]):
        return [yellow_style] * len(row)
    else:
        return [''] * len(row)

关键修正点 #

• 基于原始数据创建样式：df1.style直接从原始数据生成Styler，保证导出内容是原始数据而非样式标记。
• 行级样式逻辑：使用apply方法按行处理，接收行数据和过滤条件，返回该行对应的样式列表，实现行级背景色设置。
• 精准行匹配：通过索引+整行相等判断当前行是否在过滤结果中，也可根据需求替换为任意列匹配的逻辑。

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