Pandas groupby 和 agg by 条件

Question

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

上面的行Month对 dataframe 进行分组，并计算每个月的Status数。 有没有办法只获得Status=X的计数？ 类似于下面的错误代码：

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

本质上，我想计算每个月有多少个Status是X

Answer 1

一条短途

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

很长的路

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

Answer 2

让我们做一些不同的事情

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

Answer 3

也可以使用lambda function

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

或者

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

Answer 4

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