Pandas - return groupby 之后的数据帧

Question

我有一个 Pandas df:

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

我想按列 Name 分组，对列 No 求和，然后 return 像这样的 2 列数据框：

Name    No
A       3
B       7

我试过了：

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

但它不是 return 我想要的数据框。我无法将结果作为列添加到数据框中。

非常感谢任何帮助

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