将参数传递给groupby和pandas中的agg

Question

我想在groupby之后使用agg并传入两个参数param1和param2。我尝试了以下但失败了。这样做的正确方法是什么？感谢。

def myfun(x, param1, param2):
    #some calculations
    return result

B = A.groupby([A.index, 'time']).agg({'salary': myfun, param1, param2})

Answer 1

使用以下语法：

param1=1
param2=100
A.groupby([A.index,'time'])['salary'].agg(myfun, param1, param2)

或@ayhan建议：

A.groupby([A.index,'time']).agg({'salary':lambda x: myfunc(x, param1, param2)})