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列之间的所有独特组合,并合并它们.count列应该在合并列之间添加,结果应如下所示:

    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