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 ，請參閱文檔和筆記本 ：

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