python - 为什么 Pandas 给出 AttributeError : 'SeriesGroupBy' object has no attribute 'pct' ?

Question

我正在尝试将用户定义的函数 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'

Answer 1

有字符串'pct'，需要变量pct - 删除''的lambda函数:

aggs = {'B':pct}
print(df.groupby('A').agg(aggs))

          B
A          
1  0.333333
4  0.333333
7  0.333333