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

时间:2017-07-25 18:17:50

标签: python pandas

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

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

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

1 个答案:

答案 0 :(得分:2)

使用以下语法:

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)})