Search code examples
pythonpandasmatplotlibseabornheatmap

How do I add a title and axis labels to Seaborn Heatmap?


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

Solution

  • 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