python pandas，试图找到两列的唯一组合并在对第三列求和时合并

Question

您好，我将通过示例展示我尝试做的事情： 我从这样的数据框开始：

> pd.DataFrame({'A':['a','a','a','c'],'B':[1,1,2,3], 'count':[5,6,1,7]})
    A   B   count
0   a   1   5
1   a   1   6
2   a   2   1
3   c   3   7

我需要找到一种方法来获取 A 列和 B 列之间的所有唯一组合，并将它们合并。合并后的列之间应将计数列相加，结果应如下所示：

    A   B   count
0   a   1   11
1   a   2   1
2   c   3   7

感谢任何帮助。

Answer 1

将groupby 与聚合sum 一起使用：

print (df.groupby(['A','B'], as_index=False)['count'].sum())
   A  B  count
0  a  1     11
1  a  2      1
2  c  3      7

print (df.groupby(['A','B'])['count'].sum().reset_index())
   A  B  count
0  a  1     11
1  a  2      1
2  c  3      7