python - Pandas Groupby 和 sum 有两个变量-

Question

我正在对两个变量进行分组和求和。第二个变量是年份，对于我的示例，只有两年(2015 年和 2016 年)。对于总和的第二行，现在显示第一个变量(ID#)。如何强制显示？

代码:

totals = df.groupby(by=['id', 'year'])['sales'].sum()
print(totals)

输出样本:

1234567             2015             596407.81
                    2016            7224148.34

如何让第二行的 ID 为 1234567？

Answer 1

在groupby中使用参数as_index=False :

totals = df.groupby(by=['id', 'year'], as_index=False)['sales'].sum()
print(totals)

或reset_index :

totals = df.groupby(by=['id', 'year'])['sales'].sum().reset_index()
print(totals)

在第一列中看不到最后一个值的原因是 MultiIndex