如何在Seaborn热力图颜色条中添加标签？

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'))

我该如何给色柱添加标签？

- ishido

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 对于设置色条的其他属性（如刻度频率或格式）非常有用。

- miradulo

Answer 2

您可以使用：

ax = sns.heatmap(data.pivot_table(index='y', columns='x', values='z'))
ax.collections[0].colorbar.set_label("Hello")

- kezzos