¿Cómo agregar una etiqueta a la barra de color Seaborn Heatmap?

Si tengo los siguientes datos y 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')) 

¿Cómo agrego una etiqueta a la barra de color?

Puede configurarlo después de recogerlo de un ax , o simplemente pasar una etiqueta en cbar_kws como tal.

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

introduzca la descripción de la imagen aquí

Vale la pena señalar que cbar_kws puede ser útil para configurar otros atributos en la barra de colores, como la frecuencia de marcación o el formato.

Puedes usar:

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