通过特定公式对大熊猫进行分组

Question

请帮助我，如何将pandas groupby与总和的某列一起使用，例如，将平均值用于某列。我有一个名为“市场”的数据框：

product quantity    sell_price  profit
0   bread   2   5   10
1   bread   3   4   12
2   bread   1   3   3
3   milk    3   2   6
4   milk    2   1   2

可复制的示例：

df = pd.DataFrame({'product': ['bread', 'bread', 'bread', 'milk', 'milk'], 
                   'quantity': [2, 3, 1,3,2],
                  'sell_price': [5,4,3,2,1],
                  'profit': [10,12,3,6,2]})

像这样的桌子

[Product][quantity][sell_price][profit]
[bread][6][4][25]
[milk][5][1,5][8]

其中，sell_price是每种产品的平均值， 数量是每种产品的总和，并且 利润是每种产品的总和

我已经做了

market.groupby('product').sum('quantity,'profit').mean('sell_price')

Answer 1

使用pandas.DataFrame.groupby.agg：

new_df = df.groupby('Product').agg({'quantity':'sum', 'profit': 'sum', 'sell_price': 'mean'})
print(new_df)

输出：

         quantity  profit  sell_price
Product                              
bread           6      25         4.0
milk            5       8         1.5