Pandas 將 dataframe 列與 groupby 結果相乘

Question

我有一個 dataframe：

cust_sign_month      month         user      country      Charge_volume
2018-01              2018-01        10        DE           100 
2018-01              2018-01        22        DE           200              
2018-01              2018-02        33        IN           100              
2019-01              2019-02        42        IN           200             

我想要一個新的 col“Volume”，它將根據以下計算存儲結果：

基於分組國家和月份的“Charge_volume”列的平均值乘以用戶列

output 應如下所示：

cust_sign_month      month         user      country      Charge_volume    Volume
2018-01              2018-01        10        DE           100             1500
2018-01              2018-01        22        DE           200             3300 
2018-01              2018-02        33        IN           100             4950 
2019-01              2019-02        42        IN           200             8400

我已經嘗試了以下代碼，但它似乎不起作用：

df['Volume'] = df.groupby(['month','country'])['Charge_volume'].mean()*df['user']

Answer 1

嘗試groupby().transform ：

df['Volume'] = df['user'] * df.groupby(['month','country'])['Charge_volume'].transform('mean')