为Pandas多层索引应用样式：基于多索引层级实现索引着色

我尝试基于索引的类别为DataFrame设置单元格背景色，索引0级是「水果」、1级是「部位」，希望按水果+部位的组合为所有单元格着色。目前单元格的着色已经实现，但索引的样式设置遇到了问题。

示例代码 #

import pandas as pd

# 根据索引值动态设置样式
def style_row(row):
    color_map = {
        ('apple', 'skin'): 'red',
        ('apple', 'seed'): 'blue',
        ('banana', 'skin'): 'red',
        ('banana', 'seed'): 'blue',
    }
    fruit, part, _ = row.name
    color = color_map.get((fruit, part))
    return [f'background-color: {color}']*len(row)
    
# 样本数据
data = [
    ('apple', 'skin', 'color', 'red', 2),
    ('apple', 'skin', 'surface_area', 15, 3),
    ('apple', 'skin', 'texture', 'smooth', 1),
    ('apple', 'seed', 'color', 'grey', 3),
    ('apple', 'seed', 'weight', 4, 2),
    ('banana', 'skin', 'color', 'yellow', 3),
    ('banana', 'skin', 'surface_area', 25, 21),
    ('banana', 'skin', 'texture', 'smooth', 5),
    ('banana', 'seed', 'color', 'grey', 5),
    ('banana', 'seed', 'weight', 6, 2)
]

# 创建DataFrame
df = pd.DataFrame(data, columns=['fruit', 'part', 'property', 'value', 'observances'])
df = df.set_index(['fruit', 'part', 'property'])

# 生成带样式的DataFrame
styled_df = df.style.apply(lambda row: style_row(row), axis=1)
styled_df

当前效果 #

上述代码已实现单元格的红、蓝着色，但无法基于索引0级和1级的值为索引列设置对应样式。

问题根源 #

问题源于pandas.io.formats.style.Styler API的限制：Styler.map_index、Styler.format_index、Styler.apply_index等方法处理多层索引时，回调函数会独立调用每个层级的值，无法同时获取多个索引层级的上下文来做样式决策。

尝试过的无效方案 #

该方案试图为多层索引的每个层级补充其他层级的上下文，设置样式后再还原索引，但styled_df.index = new_multiindex的修改操作并未对styled_df.map_index(callback_func)传递的值产生影响，无法达到预期效果：

# 索引样式设置函数
def style_indices(index_val):
    if isinstance(index_val, tuple):
        fruit, part, _ = index_val
        color_map = {
            ('apple', 'skin'): 'red',
            ('apple', 'seed'): 'blue',
            ('banana', 'skin'): 'red',
            ('banana', 'seed'): 'blue',
        }
        return f'background-color: {color_map.get((fruit, part), "white")}'


# 将多层索引转换为扁平索引以便修改
flat_index = styled_df.index.to_flat_index()

# 为索引值补充上下文，供回调函数使用
new_tuples = [(a, flat_index[idx], flat_index[idx]) for idx, (a, b, _) in enumerate(styled_df.index)]
new_multiindex = pd.MultiIndex.from_tuples(new_tuples)
styled_df.index = new_multiindex # 此操作未达到预期效果！

# 为索引应用样式
styled_df = styled_df.map_index(lambda index_val: style_indices(index_val), axis=0)

# 将索引还原为原始形式
simplified_index = pd.MultiIndex.from_tuples([(a, b[1], c[2]) for a, b, c in styled_df.index])
styled_df.index = simplified_index # 此操作未达到预期效果！

styled_df

注意事项 #

• 必须保持DataFrame的多层索引，以便df.to_excel()自动应用合并居中样式。
• 解决方案需兼容.to_excel()函数，确保导出的Excel文件保留样式。

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