Jak dodać tytuł do paska koloru Seaborn Heatmap?

Question

Jeśli mam następujące dane i Seaborn termiczna:

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

Jak mogę dodać tytuł na pasku kolorów?

Answer 1

Można go ustawić potem po pobraniu go z ax, lub po prostu przejść przez wytwórnię w cbar_kws jak tak.

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

Warto zauważyć, że cbar_kws może być przydatne do ustawiania innych atrybutów na colorbar takie jak częstotliwość kleszcza lub formatowania.

Answer 2

Można użyć:

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