python - 如何在复杂的pandas groupby中绘制图表？

Question

我有一个关于行事务的 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函数的输出而言。非常感谢

Answer 1

您想要unstack，然后调用pd.DataFrame.plot:

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

您会得到一个包含两行的图表，每个类别各一条。