pandas style应用颜色掩码数组

Question

我正在尝试使用蒙版数组来应用该样式。

例如，这些是DataFrame，而掩码DataFrame有true或false。

index = pd.MultiIndex.from_product([pd.date_range('2018-01-01', '2018-01-03'), [1, 2]])
columns = pd.MultiIndex.from_product([['a', 'b', 'c'], ['x', 'y']])

df = pd.DataFrame(index=index, columns=columns, data=np.random.random((6, 6)))

mask = df.copy()
mask.loc[:, :] = False

mask.loc['2018-01-01', 'a'] = True

我使用创建的函数：

def highlight(data, color_mask):
    '''
    highlight the maximum in a Series or DataFrame
    '''
    attr = 'background-color: yellow'
    return pd.DataFrame(np.where(color_mask, attr, ''),
                        index=data.index, columns=data.columns)

但是，当我运行apply函数时，它会抱怨数据框不能被散列。

df.style.apply(highlight, mask)

我觉得我做了tutorial中提到的相同的事情，但为什么它不起作用？

感谢。