Groupby et Ajouter une colonne agrégée
df.assign(SUM=df.groupby('A').B.transform('sum'))
Arrogant Alpaca
df.assign(SUM=df.groupby('A').B.transform('sum'))