来自数据框的条形图

时间:2018-07-17 07:10:38

标签: python pandas plot

我有一个看起来像这样的数据框。

print (df)
   a      b
0  1   5896
1  1   4000
2  1  89647
3  2     54
4  2   3568
5  2  48761
6  3   5896
7  3   2800
8  3   5894

enter image description here

我想绘制一个条形图。看起来像这样。

enter image description here

我尝试使用groupby.(),但它只打印1的一个值和2的等值...

a = df_result.groupby(['column1'])['column2'].mean()
a.plot.bar()
plt.show()

将感谢您提供一些有关如何解决问题的指南,因此,我将在图表中包含所有值。

1 个答案:

答案 0 :(得分:3)

我认为需要cumcountset_indexunstack来重塑数据:

a = df.set_index(['a',df.groupby('a').cumcount()])['b'].unstack()
print (a)
      0     1      2
a                   
1  5896  4000  89647
2    54  3568  48761
3  5896  2800   5894

a.plot.bar()

graph