72

If I have the following data and 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'))

How do I add a label to the colour bar?

Adriaan
  • 17,741
  • 7
  • 42
  • 75
ishido
  • 4,065
  • 9
  • 32
  • 42

2 Answers2

126

You could set it afterwards after collecting it from an ax, or simply pass a label in cbar_kws like so. 

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

enter image description here

It is worth noting that cbar_kws can be handy for setting other attributes on the colorbar such as tick frequency or formatting.

miradulo
  • 28,857
  • 6
  • 80
  • 93
16

You can use:

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