如何在复杂的熊猫 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()

您会得到一个包含两条线的图表，每个 Category 一条。