如何在Seaborn Heatmap颜色栏中添加标签？

Question

如果我有以下数据和Seaborn热图：

import pandas as pd
data = pd.DataFrame({'x':(1,2,3,4),'y':(1,2,3,4),'z':(14,15,23,2)})

sns.heatmap(data.pivot_table(index='y', columns='x', values='z'))

如何在颜色栏中添加标签？

Answer 1

您可以在从收集数据后再进行设置，也可以像这样ax简单地传递标签cbar_kws。

import seaborn as sns
import pandas as pd
data = pd.DataFrame({'x':(1,2,3,4),'y':(1,2,3,4),'z':(14,15,23,2)})

sns.heatmap(data.pivot_table(index='y', columns='x', values='z'), 
                             cbar_kws={'label': 'colorbar title'})

值得注意的是，cbar_kws在色条上设置其他属性（例如刻度频率或格式设置）可能会很方便。