python - 基于 Pandas DataFrame 中另一列的 Sum 列

Question

我有一个像这样的 Pandas 数据帧:

>>> df = pd.DataFrame({'MONTREGL':[10,10,2222,35,200,56,5555],'SINID':['aaa','aaa','aaa','bbb','bbb','ccc','ccc'],'EXTRA':[400,400,400,500,500,333,333]})
>>> df
   MONTREGL SINID EXTRA
0        10   aaa   400
1        10   aaa   400
2      2222   aaa   400
3        35   bbb   500
4       200   bbb   500
5        56   ccc   333
6      5555   ccc   333

我想对列求和 MONTREGL对于每个 groupby SINID ...

所以我得到 2242 为 aaa 等等......我还想保留列 EXTRA 的值.

这是预期的结果:

   MONTREGL SINID EXTRA
0      2242   aaa   400
1       235   bbb   500
2      5611   ccc   333

提前感谢您的帮助!

Answer 1

我最终使用了这个脚本:

dff = df.groupby(["SINID","EXTRA"]).MONTREGL.sum().reset_index()

它适用于本次测试和生产。