Pandas group-by和sum / stored

Question

我试图将一个groupby sum查询的结果存储在一个变量中，所以我可以在以后的绘图中使用它，但每当我尝试访问它时Python都会通过错误消息。任何帮助将非常感激。

Answer 1

总和已存储在stage2数据帧中。不要在代码中调用stage2.index（），而是尝试只运行stage2。

stage1 = pd.DataFrame({'Year': [2018, 2019, 2020, 2021, 2019], 'Incidents': [1,2,3,4,5]})
stage2 = stage1.groupby(['Year']).sum()
print(stage2)

输出：

      Incidents
Year           
2018          1
2019          7
2020          3
2021          4