Pandas groupby 和 agg 按条件

Question

df.groupby(['Month']).agg({'Status' : ['count']})

上面的行将数据框分组Month并计算Status每个月的数量。有没有办法只得到一个计数在哪里Status=X？类似于下面不正确的代码：

df.groupby(['Month']).agg({'Status' == 'X' : ['count']})

从本质上讲，我想多少计数Status是X每月。

Answer 1

也可以使用lambda函数

df.groupby('Month').agg(lambda x: (x=='X').sum())

或者

df.groupby('Month').Status.agg(lambda x: (x=='X').sum())

Answer 2

一条捷径

(df.Status == 'X').groupby(df.Month).sum()

很长的路

df.where(df.Status == 'X').groupby('Month').Status.count()

Answer 3

让我们做点不一样的

pd.crosstab(df.Month,df.Status)['X']