次のデータと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'))
カラーバーにラベルを追加するにはどうすればよいですか?
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
は、ティック周波数やフォーマットなど、カラーバーの他の属性を設定するのに便利です。
次を使用できます。
ax = sns.heatmap(data.pivot_table(index='y', columns='x', values='z'))
ax.collections[0].colorbar.set_label("Hello")