python - 按收入总和和相应的最高贡献月份分组 - Pandas

Question

我有一个账单明细数据集，我想根据产品总值(value)的总和对产品进行分组，另外我想要一个列来指示为相应产品产生最多收入的月份

数据集:

Bill_Id  Month  Product_Id  Net_Value
 1        1       20          100
 2        1       20          100
 3        2       20          100
 4        1       30          200
 5        2       30          200
 6        2       30          200

Desired_Result

Product_Id  Total_revenue  Top_Month
   20          300            1
   30          600            2

这只是一个示例数据集，我有全年的交易数据

Answer 1

使用 aggfunc=sum 旋转数据框，然后使用 sum 和 idxmax 沿列轴查找总收入和最高的月份对总收入的贡献，最后 concat 沿列轴的各个组件得到结果

s = df.pivot_table('Net_Value', 'Product_Id', 'Month', aggfunc='sum')
pd.concat([s.sum(1), s.idxmax(1)], axis=1, keys=['Total_revenue', 'Top_Month'])

---

            Total_revenue  Top_Month
Product_Id                          
20                    300          1
30                    600          2