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')