df has the following cols, df['Account ID'] and df['Revenue']. df['Account ID'] has duplicate values. Tried the following but it generates empty col:

df['Total_Revenue'] = df.groupby('Account ID')['Revenue'].sum()

thanks for the help.

CodePudding user response：

Use assign method to create new column then groupby transform to not collapse the aggregation.

Code:

df.assign(total_revenue=lambda df_: df_.groupby("Account ID").Revenue.transform("sum"))

Documentation:

• Pandas - assign

• Pandas - transform