如何基于相邻单元格值为Pandas DataFrame特定单元格设置样式？

核心思路 #

利用Pandas Styler的apply方法结合自定义样式函数，精准指定需要处理的单元格范围，通过对比当前单元格与右侧相邻单元格的值来设置高亮样式，完全避开不需要处理的表头或其他区域。

实现步骤与代码示例 #

1. 数据准备：将NumPy数组转换为DataFrame

import pandas as pd
import numpy as np

# 你的示例数据
data = np.array([['A', 1, 1, 2, 1],
                 ['B', 1, 2, 3, 3],
                 ['C', 3, 3, 1, 1]])
df = pd.DataFrame(data)

2. 自定义高亮函数
   这个函数接收指定范围的子DataFrame，生成对应的样式矩阵，仅对右侧值不同的单元格设置高亮：

def highlight_different_right(sub_df):
    # 初始化样式矩阵，默认无样式
    styles = pd.DataFrame('', index=sub_df.index, columns=sub_df.columns)
    
    # 遍历除最后一列外的所有列（最后一列无右侧单元格）
    for col in sub_df.columns[:-1]:
        # 标记当前列与右侧列值不同的单元格
        diff_mask = sub_df[col] != sub_df[col + 1]
        styles.loc[diff_mask, col] = 'background-color: #ffff99'  # 浅黄色高亮
    
    return styles

3. 应用样式并导出Excel
   通过subset参数指定要处理的单元格范围（比如示例中需要处理行0-2、列1-3，避开表头列和无右侧的最后一列）：

# 精准指定处理范围：行索引0到2，列索引1到3
styled_df = df.style.apply(highlight_different_right, subset=pd.IndexSlice[0:3, 1:4])

# 导出到Excel，保留样式
styled_df.to_excel('highlighted_result.xlsx', index=False)

关键说明 #

• 范围控制：pd.IndexSlice可以灵活指定行和列的范围，比如你的表头在第0行，就把subset的行范围设为1:，完全不影响表头。
• 边界处理：自动跳过最后一列，因为该列没有右侧相邻单元格，无需判断。
• 样式自定义：可以修改background-color的值来调整高亮颜色，比如换成lightcoral实现红色高亮。

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