Pandas 设置元素样式依赖于另一个数据框 mith 多索引

Question

我之前问过问题 Pandas set element style dependent on another dataframe，我有一个可行的解决方案，但现在我试图将它应用于具有多索引的数据框，但出现错误，我不明白。

问题

我有一个 pandas df 和随附的布尔矩阵。我想根据布尔矩阵突出显示 df。

数据

import pandas as pd
import numpy as np
from datetime import datetime

date = pd.date_range(start = datetime(2016,1,1), end = datetime(2016,2,1), freq = "D")
i = len(date)
dic = {'X':pd.DataFrame(np.random.randn(i, 2),index = date, columns = ['A','B']),
       'Y':pd.DataFrame(np.random.randn(i, 2),index = date, columns = ['A','B']),
       'Z':pd.DataFrame(np.random.randn(i, 2),index = date, columns = ['A','B'])}
df = pd.concat(dic.values(),axis=1,keys=dic.keys())


boo =  [True, False]
bool_matrix = {'X':pd.DataFrame(np.random.choice(boo, (i,2), p=[0.3,.7]), index = date, columns = ['A','B']),
               'Y':pd.DataFrame(np.random.choice(boo, (i,2), p=[0.3,.7]), index = date, columns = ['A','B']),
           'Z':pd.DataFrame(np.random.choice(boo, (i,2), p=[0.3,.7]), index = date, columns = ['A','B'])}

bool_matrix =pd.concat(bool_matrix.values(),axis=1,keys=bool_matrix.keys())

我尝试的解决方案

def highlight(value):
    return 'background-color: green' 
my_style = df.style
for column in df.columns:
    for i in df[column].index:
        data = bool_matrix.loc[i, column]
        if data: 
            my_style = df.style.use(my_style.export()).applymap(highlight, subset = pd.IndexSlice[i, column])

my_style

结果

以上抛出一个AttributeError: 'Series' object has no attribute 'applymap'

我不明白什么是系列回归。这是我正在子集化的单个值，此解决方案适用于非多索引 df，如下所示。

没有多索引

import pandas as pd
import numpy as np
from datetime import datetime
np.random.seed(24)
date = pd.date_range(start = datetime(2016,1,1), end = datetime(2016,2,1), freq = "D")
df = pd.DataFrame({'A': np.linspace(1, 100, len(date))})
df = pd.concat([df, pd.DataFrame(np.random.randn(len(date), 4), columns=list('BCDE'))],
               axis=1)

df['date'] = date
df.set_index("date", inplace = True)

boo =  [True, False]
bool_matrix = pd.DataFrame(np.random.choice(boo, (len(date), 5),p=[0.3,.7]), index = date,columns=list('ABCDE'))

def highlight(value):
    return 'background-color: green' 
my_style = df.style
for column in df.columns:
    for i in bool_matrix.index:
        data = bool_matrix.loc[i, column]
        if data: 
            my_style = df.style.use(my_style.export()).applymap(highlight, subset = pd.IndexSlice[i,column])
my_style

文档

docs 参考了 CSS 类并说 "Index label cells include level where k is the level in a MultiIndex." 我显然索引这个错误，但我不知道如何继续。

Answer 1

很高兴有一个可运行的例子。

您可以使用 df.style.apply(..., axis=None) 将突出显示方法应用于整个数据框。

用你的 df 和 bool_matrix，试试这个：

def highlight(value):
    d = value.copy()
    for c in d.columns:
        for r in df.index:
            if bool_matrix.loc[r, c]:
                d.loc[r, c] = 'background-color: green'
            else:
                d.loc[r, c] = ''
    return d

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

或者为了简化代码，你可以试试：

def highlight(value):
    return bool_matrix.applymap(lambda x: 'background-color: green' if x else '')

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

希望这是你需要的。

Pandas 设置元素样式依赖于另一个数据框 mith 多索引

Pandas set element style dependent on another dataframe mith multi index

python

multi-index

pandas

pandas-styles

问题

数据

我尝试的解决方案

结果

没有多索引

文档