如何按两列分组并创建子图循环？

Question

我在这个结构中有一个大数据框（df）：    year        person      purchase    2016        Peter       0    2016        Peter       223820    2016        Peter       0    2017        Peter       261740    2017        Peter       339987    2018        Peter       200000    2016        Carol       256400    2017        Carol       33083820    2017        Carol       154711    2018        Carol       3401000    2016        Frank       824043    2017        Frank       300000    2018        Frank       214416259    2018        Frank       4268825    2018        Frank       463080    2016        Rita        0要查看每个人每年花费多少，我会按年和人进行分组，这给了我想要的东西。代码：df1 = df.groupby(['person','year']).sum().reset_index()如何创建一个循环来为每个人创建包含他/她每年在购买上花费的子图？所以每个人都有一个子图，其中 x = 年份和 y = 购买。我已经尝试了很多不同的东西在这里解释，但似乎没有一个工作。

Answer 1

您可以执行pivot_table或groupby().sum().unstack('person')然后plot：

(df.pivot_table(index='year', 

                columns='person', 

                values='purchase', 

                aggfunc='sum')

    .plot(subplots=True)

);

或者

(df.groupby(['person','year'])['purchase']

   .sum()

   .unstack('person')

   .plot(subplots=True)

);

输出：