How to add a label to Seaborn Heatmap color bar?

python heatmap seaborn
ishido picture ishido · Feb 7, 2017 · Viewed 26.5k times · Source

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?

Answer

miradulo picture miradulo · Feb 7, 2017

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.

Related questions

Make the size of a heatmap bigger with seaborn

I create a heatmap with seaborn df1.index = pd.to_datetime(df1.index) df1 = df1.set_index('TIMESTAMP') df1 = df1.resample('30min').mean() ax = sns.heatmap(df1.iloc[:, 1:6:], annot=True, linewidths=.5) But the probleme is when there is lot …

Read More
Changing the rotation of tick labels in Seaborn heatmap

I'm plotting a heatmap in Seaborn. The problem is that I have too many squares in my plot so the x and y labels are too close to each other to be useful. So I'm creating a list of xticks …

Read More
Custom Annotation Seaborn Heatmap

I'm using Seaborn in Python to create a Heatmap. I'm able to annotate the cells with the values passed in, but I'd like to add annotations that signify what the cell means. For example, instead of merely seeing 0.000000, I'd like …

Read More