python - Pandas groupby agg - 如何获得计数？

Question

我正在尝试获取指标的总和、均值和计数

df.groupby(['id', 'pushid']).agg({"sess_length": [ np.sum, np.mean, np.count]})

但我得到“模块‘numpy’没有属性‘count’”，我尝试了不同的方式来表达计数函数，但无法让它工作。我如何将汇总记录数与其他指标一起计算？

Answer 1

您可以使用字符串而不是函数，如下所示:

df = pd.DataFrame(
    {"id": list("ccdef"), "pushid": list("aabbc"), 
     "sess_length": [10, 20, 30, 40, 50]}
)

df.groupby(["id", "pushid"]).agg({"sess_length": ["sum", "mean", "count"]})

哪些输出:

           sess_length
                   sum mean count
 id pushid
 c  a               30   15     2
 d  b               30   30     1
 e  b               40   40     1
 f  c               50   50     1