"如何為 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可以方便地設置顏色欄上的其他屬性，例如刻度頻率或格式。

Answer 2

您可以使用：

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