我想在groupby之后使用agg并传入两个参数param1和param2。我尝试了以下但失败了。这样做的正确方法是什么?感谢。
def myfun(x, param1, param2):
#some calculations
return result
B = A.groupby([A.index, 'time']).agg({'salary': myfun, param1, param2})
答案 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)})