聚合函数作为参数

Question

我需要写一个函数：

def group_and_aggregate_data(df:pd.DataFrame, group_by_column:str, agg_func) -> pd.DataFrame

按城市名称对我的 Excel 数据进行分组，并应用作为参数传递的

agg_func

。我试过这个：

def group_and_aggregate_data(dataframe, cond, func):
    df_bycity = dataframe.groupby(cond).func()
    return df_bycity

但它在Python中不起作用。

Answer 1

你可以用 df.agg() 来尝试一下吗:

def group_and_aggregate_data(dataframe, cond, func):
    df_bycity = dataframe.groupby(cond).agg(func)
    return df_bycity