在熊猫中注释条形图

时间:2017-12-02 03:03:40

标签: python python-3.x pandas matplotlib jupyter-notebook

我有一个条形图,显示2015年的一个条形图和2016年的一个条形图。

  • x轴显示药物名称
  • y轴表示规定的数量

我想在条形图上显示注释,以便更明确规定的数字。我尝试过以下但没有任何反应?

有谁知道我哪里出错了?

这是我的代码

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

2 个答案:

答案 0 :(得分:1)

尝试将前两行移至plt.show()之前。当前代码在定义之前引用ax

答案 1 :(得分:1)

你认为这有助于:

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

修改

如果您想要更好的风格,可以在开头添加

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

并使用ax.legend(["2015", "2016"])修改ax.legend(["2015", "2016"], frameon=False)。有关样式类型plt.style.available

的完整列表