Pandas - 在groupby

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