How do I add a title to Seaborn Heatmap?

Ammar Akhtar picture Ammar Akhtar · Sep 22, 2015 · Viewed 56.7k times · Source

I want to add a title to a seaborn heatmap. Using Pandas and iPython Notebook

code is below,

a1_p = a1.pivot_table( index='Postcode', columns='Property Type', values='Count', aggfunc=np.mean, fill_value=0)

sns.heatmap(a1_p, cmap="YlGnBu")

the data is pretty straight forward:

In [179]: a1_p

Out [179]:
Property Type   Flat    Terraced house  Unknown
Postcode            
E1  11  0   0
E14 12  0   0
E1W 6   0   0
E2  6   0   0

Answer

areuexperienced picture areuexperienced · Sep 22, 2015

heatmap is an axes-level function, so you should be able to use just plt.title or ax.set_title:

%matplotlib inline
import numpy as np
import os
import seaborn as sns
import matplotlib.pyplot as plt

data = np.random.randn(10,12)

ax = plt.axes()
sns.heatmap(data, ax = ax)

ax.set_title('lalala')
plt.show()

enter image description here