python - pandas groupby 多列

Question

我有一个数据集，其中包含状态代码及其状态。

  code  status
1   AZ  a
2   CA  b
3   KS  c
4   MO  c
5   NY  d
6   AZ  d
7   MO  a
8   MO  b
9   MN  b
10  NV  a
11  NV  e
12  MO  f
13  NY  a
14  NY  a
15  NY  b

我想过滤掉这个代码只包含 a 的数据集状态并计算他们有多少。示例输出将是，

  code  status  
1   AZ  a   
2   MO  a   
3   NY  a   

    AZ =1   MO = 1  NY =2

我用过df.groupyby("code").loc[df.status == 'a']但没有任何运气。任何帮助表示赞赏!

Answer 1

让我们首先过滤数据帧中的 a，然后进行 groupby 和 count。

df[df.status == 'a'].groupby('code').size()

输出:

code
AZ    1
MO    1
NV    1
NY    2
dtype: int64