在Pandas Group中对列进行求和

Question

我想按A列进行分组，然后求和C列并立即将结果返回到数据框中。我知道我需要使用groupby，而且我知道我需要使用sum，但我无法弄清楚如何让这些函数在一行代码中无缝交互。

有

A B  C 
0 x text 3 
1 x text 7 
2 y text 5 

要上groubpy

A B  C D 
0 x text 3 10 
1 x text 7 10 
2 y text 5 5 

Answer 1

呼叫transform添加聚集列回原来的DF：

In [28]: 
df['D'] = df.groupby('A')['C'].transform('sum') 
df 

Out[28]: 
    A  B C D 
0 x text 3 10 
1 x text 7 10 
2 y text 5 5 

transform返回一系列与其指数对齐到原来的DF，所以你可以添加它作为一个新的列