web-dev-qa-db-ja.com

Seaborn Heatmapカラーバーにラベルを追加する方法は?

次のデータとSeaborn Heatmapがある場合:

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

カラーバーにラベルを追加するにはどうすればよいですか?

23
ishido

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

enter image description here

cbar_kwsは、ティック周波数やフォーマットなど、カラーバーの他の属性を設定するのに便利です。

38
miradulo

次を使用できます。

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