带有嵌套行的 Seaborn 热图

Question

给定以下数据框和数据透视表：

import pandas as pd
df=pd.DataFrame({'A':['a','a','a','a','a','b','b','b','b'],
                 'B':['x','y','z','x','y','z','x','y','z'],
                 'C':['a','b','a','b','a','b','a','b','a'],
                 'D':[7,5,3,4,1,6,5,3,1]})
table = pd.pivot_table(df, index=['A', 'B','C'],aggfunc='sum')
table

            D
A   B   C   
a   x   a   7
        b   4
    y   a   1
        b   5
    z   a   3
b   x   a   5
    y   b   3
    z   a   1
        b   6

我想创建一个按索引 A 和 B 划分的热图，如下所示：

有可能吗？

Answer 1

您可以在jupyter 笔记本中使用Styler，参见docs 和notebook：

import seaborn as sns
import pandas as pd

df=pd.DataFrame({'A':['a','a','a','a','a','b','b','b','b'],
                 'B':['x','y','z','x','y','z','x','y','z'],
                 'C':['a','b','a','b','a','b','a','b','a'],
                 'D':[7,5,3,4,1,6,5,3,1]})
table = pd.pivot_table(df, index=['A', 'B','C'],aggfunc='sum')
table


cm = sns.light_palette("blue", as_cmap=True)
s = df.reset_index().style.background_gradient(cmap=cm)
s