python pandas groupby 然后计算满足条件的行

Question

我正在尝试对 id 列 进行分组，以便可以显示 col1 中等于 1 的行数。

df:

id col1 col2 col3
a   1     1    1
a   0     1    1
a   1     1    1
b   1     0    1

我的代码:

df.groupby(['id'])[col1].count()[1]

我得到的输出是2。它没有向我显示其他id(例如b)的值。

我想要:

id col1
a   2    
b   1 

如果可能的话，每个id的总行数也可以显示为新列吗？

示例:

id col1 total
a   2    3
b   1    1

Answer 1

假设 col1 中只有 1 和 0，则可以使用 agg:

df.groupby('id', as_index=False)['col1'].agg({'col1': 'sum', 'total': 'count'})

#  id   total   col1
#0  a       3      2
#1  b       1      1