python - 将多个函数应用于 pandas groupby apply 返回多个数据帧？

Question

当前我正在三个 groupby 中执行此操作，每个 apply 返回一个数据帧:

 fd1 = df.groupby('loan_id').apply(build_feature_set_1)
 fd2 = df.groupby('loan_id').apply(build_feature_set_2)
 fd3 = df.groupby('loan_id').apply(build_feature_set_3)

这些函数需要访问多个列，因此我无法使用agg。本质上，我希望能够在一次数据扫描中完成此操作，即一次分组。可能吗？

Answer 1

您可以尝试这种方法:-

fn_list = [build_feature_set_1, build_feature_set_2, build_feature_set_3]

grouped_df = df.groupby('loan_id')

df1, df2, df3 = list(map(lambda x: grouped_df.apply(x), fn_list))