Как добавить заголовок к цветовой гамме Seaablot Heatmap?

Question

Если у меня есть следующие данные и Сиборн 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')) 

Как добавить название цветной полосы?

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 может быть удобно для настройки других атрибутов на Colorbar, такие как частота тиков или форматирования.

Answer 2

Вы можете использовать:

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