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

Question

I want to use agg after groupby and pass in two parameters, param1 and param2. I tried the following but failed. What is the correct way to do that? Thanks.

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

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

Answer 1

Use this syntax:

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

or as @ayhan suggests:

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