我有一个熊猫数据框如下
A, B
----
a, 2
a, 5
a, 6
b, 1
b, 2
我想对列进行分组A并汇总列中的值B，并将其附加为另一列并创建以下数据框
A, B, SUM
--------
a, 2, 13
a, 5, 13
a, 6, 13
b, 1, 3
b, 2, 3
如何在熊猫中做到这一点？
使用 transform
df.assign(SUM=df.groupby('A').B.transform('sum'))
Out[15]:
A B SUM
0 a 2 13
1 a 5 13
2 a 6 13
3 b 1 3
4 b 2 3
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句