如何使用df.style高亮Pandas DataFrame的非主对角线元素

原代码逻辑拆解 #

原代码通过df.style.apply实现自定义样式，核心逻辑如下：

• 创建一个和原DataFrame结构完全一致的空样式容器diag_mask，所有单元格初始值为空字符串（表示无样式）。
• 获取DataFrame行数和列数的较小值min_axis，避免索引越界。
• 通过iloc[range(min_axis), range(min_axis)]选中主对角线（行索引与列索引相等）的单元格，给这些单元格设置background-color: yellow样式。
• 返回样式容器，由df.style.apply应用到原表格上。

修改代码实现目标对角线高亮 #

要高亮从第二列起始的对角线（对应元素1、4、9的位置），只需调整选中单元格的索引规则，具体代码如下：

import numpy as np
import pandas as pd

df = pd.DataFrame({'a':[1,2,3,4],'b':[1,3,5,7],'c':[1,4,7,10],'d':[1,5,9,11]})

def style_offset_diag(data):
    # 创建与原DataFrame结构一致的空样式容器
    diag_mask = pd.DataFrame("", index=data.index, columns=data.columns)
    # 计算可选中的最大行索引：列索引=行索引+1，避免列索引越界
    max_row_idx = len(data.index) - 2
    # 选中目标对角线的单元格并设置样式
    diag_mask.iloc[range(max_row_idx+1), range(1, max_row_idx+2)] = 'background-color: yellow'
    return diag_mask

df.style.apply(style_offset_diag, axis=None)

通用调整方法 #

如果需要高亮其他偏移量的对角线，只需修改索引范围：

• 高亮列索引比行索引大n的对角线：列起始索引设为n，行索引范围为0到len(data.index)-n-1，列索引范围为n到len(data.index)。
• 高亮列索引比行索引小n的对角线：行起始索引设为n，行索引范围为n到len(data.index)，列索引范围为0到len(data.index)-n。

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