CODEWITHSUNDEEP
pythonpandasmatplotlibtextsubplot
jonas
jonas

Reputation: 13969

Annotate matplotlib subplot with values

I would like to annotate each barplot with the value on top each bar. I have found this excellent answer to a single plot Adding value labels on a matplotlib bar chart , however I can not figure it out with subplots.

Attached is an simplified example 

import pandas as pd
import matplotlib as mpl
from matplotlib.colors import LinearSegmentedColormap
from matplotlib.lines import Line2D


countries = ['France','Spain','Sweden','Germany','Finland','Poland','Italy',
             'United Kingdom','Romania','Greece','Bulgaria','Hungary',
             'Portugal','Austria','Czech Republic','Ireland','Lithuania','Latvia',
             'Croatia','Slovakia','Estonia','Denmark','Netherlands','Belgium']
extensions = [547030,504782,450295,357022,338145,312685,301340,243610,238391,
              131940,110879,93028,92090,83871,78867,70273,65300,64589,56594,
              49035,45228,43094,41543,30528]
populations = [63.8,47,9.55,81.8,5.42,38.3,61.1,63.2,21.3,11.4,7.35,
               9.93,10.7,8.44,10.6,4.63,3.28,2.23,4.38,5.49,1.34,5.61,
               16.8,10.8]
life_expectancies = [81.8,82.1,81.8,80.7,80.5,76.4,82.4,80.5,73.8,80.8,73.5,
                    74.6,79.9,81.1,77.7,80.7,72.1,72.2,77,75.4,74.4,79.4,81,80.5]
data = {'extension' : pd.Series(extensions, index=countries), 
        'population' : pd.Series(populations, index=countries),
        'life expectancy' : pd.Series(life_expectancies, index=countries)}

df = pd.DataFrame(data)
df = df.sort('life expectancy')

fig, axes = plt.subplots(nrows=3, ncols=1)
for i, c in enumerate(df.columns):
    df[c].plot(kind='bar', ax=axes\[i\], figsize=(12, 10), title=c)
plt.savefig('EU1.png', bbox_inches='tight')]

enter image description here

Upvotes: 0

Views: 1154

Answers (1)

Hongpei
Hongpei

Reputation: 697

It's actually similar, but here in your case it's axes[i] instead of ax in the original answer.

fig, axes = plt.subplots(nrows=3, ncols=1)
for i, c in enumerate(df.columns):
    df[c].plot(kind='bar', ax=axes[i], figsize=(12, 12), title=c)

    # here it's almost the same with your linked answer
    rects = axes[i].patches
    labels = df[c].values
    for rect, label in zip(rects, labels):
        height = rect.get_height()
        axes[i].text(rect.get_x() + rect.get_width() / 2, height + 5, label,
                ha='center', va='bottom')

fig.tight_layout() # to avoid overlapping

plt.savefig('EU1.png', bbox_inches='tight')

Note you may still need some fine tuning of y axis to avoid label go above the subplot.

Upvotes: 1

Related Questions