0

I have a bar chart that displays 2 bars one for 2015 and one for 2016.

  • x axis shows drug name
  • y axis shows amount prescribed

I would like to display annotation on the bars just to make it more clear the number prescribed. I have tried the following but nothing happens?

Does anyone know where I am going wrong?

Here is my code

for p in ax.patches:
ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005))

fig = plt.figure() # Create matplotlib figure

ax = fig.add_subplot(111) # Create matplotlib axes

width = 0.4

df.Occurance_x.plot(kind='bar', color='purple', ax=ax, width=width, position=1)
df.Occurance_y.plot(kind='bar', color='blue', ax=ax, width=width, position=0)

ax.set_ylabel('Amount Prescribed in 1 year')
x_labels = df.VTM_NM
ax.set_xticklabels(x_labels, rotation=30)
plt.legend(['2016', '2017'], loc='upper right')
ax.set_title('BNF Chapter 1 Top 5 drugs prescribed')


plt.show()
Sarah
  • 3
  • 3
  • 7

2 Answers2

2

Do you think that this can help:

import pandas as pd
df = pd.DataFrame({"date_x":[2015]*5,
                   "Occurance_x":[2994, 2543, 2307, 1535, 1511],
                   "VTM_NM":["Not Specified", "Mesalazine", "Omeprazole",
                             "Esomeprazole", "Lansoprazole"],
                   "date_y":[2016]*5,
                   "Occurance_y":[3212, 2397, 2370, 1516, 1547]})

ax = df[["VTM_NM","Occurance_x", "Occurance_y"]].plot(x='VTM_NM', 
                                                      kind='bar', 
                                                      color=["g","b"],
                                                      rot=45)
ax.legend(["2015", "2016"]);
for patch in ax.patches:
    bl = patch.get_xy()
    x = 0.5 * patch.get_width() + bl[0]
    # change 0.92 to move the text up and down
    y = 0.92 * patch.get_height() + bl[1] 
    ax.text(x,y,"%d" %(patch.get_height()),
            ha='center', rotation='vertical', weight = 'bold')

plot

Edit

If you then want a nicer style you can add this at the beginning

from matplotlib import pyplot as plt
plt.style.use('fivethirtyeight')

and modify ax.legend(["2015", "2016"]) with ax.legend(["2015", "2016"], frameon=False). For a full list of style type plt.style.available

rpanai
  • 12,515
  • 2
  • 42
  • 64
1

Try moving the first two lines to just before plt.show(). The current code references ax before defining it.

Peter Leimbigler
  • 10,775
  • 1
  • 23
  • 37