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

1 minuto de lectura

avatar de usuario
ishido

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 colores?

avatar de usuario
miradulo

Puede configurarlo después de recogerlo de un axo simplemente pasar una etiqueta en cbar_kws al igual que.

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

ingrese la descripción de la imagen aquí

Cabe resaltar que cbar_kws puede ser útil para establecer 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")

¿Ha sido útil esta solución?