熊猫-分组后返回数据框

Question

我有一只熊猫df：

Name    No
A       1
A       2
B       2
B       2
B       3

我想按列Name，求和列分组No，然后返回一个2列的数据框，如下所示：

Name    No
A       3
B       7

我试过了：

df.groupby(['Name'])['No'].sum()

但它不会返回我想要的数据框。我不能将结果作为列添加到数据框。

非常感谢任何帮助

Answer 1

将参数添加as_index=False到groupby：

print (df.groupby(['Name'], as_index=False)['No'].sum())
  Name  No
0    A   3
1    B   7

或致电reset_index：

print (df.groupby(['Name'])['No'].sum().reset_index())
  Name  No
0    A   3
1    B   7