pandas 中的聚合和计数

Question

对于下面的df

group  participated
A           1
A           1
B           0
A           0
B           1
A           1
B           0
B           0

我想计算组列中每个值的参与列中的值总数（groupby-count），然后计算每个组中有多少个 1

类似

group tot_participated   1s
A         4              3
B         4              1

我知道第一部分很简单，可以通过简单的方式完成

grouped_df=df.groupby('group').count().reset_index()

无法理解第二部分。任何帮助将不胜感激！

Answer 1

您可以使用如下聚合跟随 groupby：

grp_df = df.groupby('group', as_index=False).agg({'participated':['count','sum']})
grp_df.columns = ['group','tot_participated','1s']
grp_df.head()

在同一列上使用具有多个聚合函数的 .agg 需要注意的是，会创建一个多列索引。这可以通过在第 2 行中重置列名来解决。