如何为 Seaborn Heatmap 彩条添加标签？

【问题标题】：How to add a label to Seaborn Heatmap color bar?如何为 Seaborn Heatmap 彩条添加标签？
【发布时间】：2017-06-24 20:05:20
【问题描述】：

如果我有以下数据和 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'))

如何为颜色条添加标签？

【问题讨论】：

标签： python heatmap seaborn

【解决方案1】：

你可以使用：

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

【讨论】：

【解决方案2】：

您可以在从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 可以方便地设置颜色栏上的其他属性，例如刻度频率或格式。

【讨论】：