Pandas DataFrame条件格式问题：按指定规则高亮目标列

需求说明 #

需要实现：

• 当A列值为'x'时，高亮该行的B列
• 当A列值为'y'时，高亮该行的C列
• 其他情况无高亮

问题分析 #

原代码通过行级高亮整行+单独处理列高亮的方式无法满足需求，因为高亮规则依赖每行A列的值与对应列的关联，需要在同一个行遍历逻辑中完成判断。

解决方案 #

编写一个行级样式函数，在遍历每行时根据A列的值，仅为目标列设置背景色：

import pandas as pd

df = pd.DataFrame([['x',3,1],['x',2,2],['y',4,6]], columns=list("ABC"))

def highlight_target_col(row):
    # 初始化所有列为无背景色
    styles = [''] * len(row)
    a_value = row['A']
    if a_value == 'x':
        # 找到B列的索引，设置背景色
        b_idx = row.index.get_loc('B')
        styles[b_idx] = 'background-color: #BAFFC9'
    elif a_value == 'y':
        # 找到C列的索引，设置背景色
        c_idx = row.index.get_loc('C')
        styles[c_idx] = 'background-color: #BAFFC9'
    return styles

# 应用样式并导出到Excel
styled_df = df.style.apply(highlight_target_col, axis=1)
styled_df.to_excel('highlighted_data.xlsx', index=False)

代码解释 #

1. highlight_target_col函数接收每行数据，先初始化一个空样式列表，长度与列数一致
2. 根据当前行A列的值，判断需要高亮的列：
   • 若为'x'，获取B列的索引位置，为该位置设置背景色样式
   • 若为'y'，获取C列的索引位置，为该位置设置背景色样式
3. 返回样式列表，apply方法通过axis=1按行应用该函数
4. 最后调用to_excel导出带格式的Excel文件，index=False避免导出索引列

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