Python Pandas - 使用 function 计算最小值、最大值和平均值

Question

我正在尝试创建一个名为 sales_recap 的 function 来计算销售的最小值、最大值和平均值。

def sales_recap(sales)

例如，如果我有一个 dataframe 作为这样的输入：

sales = pd.DataFrame({
    'Type':['Electronics', 'Electronics', 'Electronics', 'Home_Decor', 'Home_Decor', 'Kids', 'Kids, 'Kids],
    'Price':[100,100,70,60,80,50,50,50]})

如果我运行 function

sales_recap(sales)

它应该给出这样的结果

recap = pd.DataFrame({
    'Type':['Electronics', 'Home_Decor','Kids'],
    'Min_Price':[70,60,50],
    'Average':[90,70,50],
    'Max_Price':[100,80,50]
})

我是 python 的新手，我仍然不知道如何使用 function 执行此操作，以便我可以将 function 与其他输入一起使用。 谢谢你： ：）

Answer 1

使用groupby_agg ：

>>> sales.groupby('Type').agg(Min_Price=('Price', 'min'),
                              Average=('Price', 'mean'),
                              Max_Price=('Price', 'max')).reset_index()

          Type  Min_Price  Average  Max_Price
0  Electronics         70     90.0        100
1   Home_Decor         60     70.0         80
2         Kids         50     50.0         50