pandas使用条件设置单元格颜色(在多索引中)

import pandas as pd
import numpy as np

# 创建多层级行和列索引
index = pd.MultiIndex.from_product([['A', 'B', 'C'], ['x', 'y']])
columns = pd.MultiIndex.from_product([['I', 'II'], ['1', '2']])

# 创建随机数据填充DataFrame
data = np.random.randn(6, 4)

# 创建DataFrame
df = pd.DataFrame(data, index=index, columns=columns)

# 定义函数来设置单元格的颜色
def color_negative_red(val):
    color = 'red' if val < 0 else 'black'
    return 'color: %s' % color

# 使用style.applymap()方法将函数应用于DataFrame的单元格
styled_df = df.style.applymap(color_negative_red)

styled_df.to_excel('styled_dataframe.xlsx', engine='openpyxl')