如何在复杂的熊猫组中绘制图形?

时间:2018-03-20 00:33:37

标签: python pandas matplotlib

我有一个关于rows-transaction的DataFrame,如:

 Month        Category       Quantity
 3            A              1 
 3            A              1
 3            B              1  
 4            A              1
 4            B              1
 4            B              1

我使用groupby函数来计算每个月中每个类别的数量

df.groupby(['Month','Category']['Quantity'].count()

结果:

Month   Category    
3       A           2
        B           1
4       A           1
        B           2 

我不知道如何绘制类别的数量图(A& B应该用不同的线表示),x轴是groupby函数输出的时间。非常感谢

1 个答案:

答案 0 :(得分:2)

您需要unstack,然后致电pd.DataFrame.plot

df.groupby(['Month','Category'])['Quantity'].count().unstack().plot()
plt.show()

您会得到一个包含两行的图表,每Category一行。