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
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()