将某些值相加

Question

我使用的是python，有一个dataframe（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

使用^{}：

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