我正在尝试将用户定义的函数pct
传递给Pandas agg
方法,如果我仅传递该函数,但当我使用字典格式来定义该函数时却不传递,则该方法有效功能。有人知道为什么吗?
import pandas as pd
df = pd.DataFrame([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]],
columns=['A', 'B', 'C'])
pct = lambda x: len(x)/len(df)
df.groupby('A').agg(pct)
按预期返回
B C
A
1 0.333333 0.333333
4 0.333333 0.333333
7 0.333333 0.333333
但是
aggs = {'B':['pct']}
df.groupby('A').agg(aggs)
返回以下错误:
AttributeError: 'SeriesGroupBy' object has no attribute 'pct'
答案 0 :(得分:3)
有字符串'pct'
,需要变量pct
-通过删除''
的lambda函数:
aggs = {'B':pct}
print(df.groupby('A').agg(aggs))
B
A
1 0.333333
4 0.333333
7 0.333333