将某些值连续汇总[重复]

Question

我在 python 中工作，并且有一个数据框（df），其中包含“COUNTERPARTYNAME”列。此列中有不同的交易对手，例如“A”、“B”、“C”等……还有另一列称为 MTM，由数字组成。

COUNTERPARTYNAME                   NPV          
0     A                            90                

1     B                            85                    

2     A                            130                     

3     C                            90    

4     B                            105

5     A                            75
...

...

基本上，我希望在我的数据框中添加一列，该列将包含该行中相应 COUNTERPARTYNAME 的所有 NPV 的总和。

我的预期结果是这样的：

    COUNTERPARTYNAME               NPV                 SUM(NPV)      
0     A                            90                  295

1     B                            85                  190  

2     A                            130                 295 

3     C                            90                  90

4     B                            105                 190

5     A                            75                  295
...

谢谢！

Answer 1

使用GroupBy.transform:

df['SUM(NPV)'] = df.groupby('COUNTERPARTYNAME')['NPV'].transform('sum')