python – 分组,大熊猫

select df.id, count(distinct airports) as num
from df
group by df.id
having count(distinct airports) > 3

我试图在Python熊猫中做相同的上述操作.我尝试过滤器,nunique,agg的不同组合,没有任何工作.有什么建议？

例如：
DF

df   
id     airport
1      lax
1      ohare
2      phl
3      lax
2      mdw
2      lax
2      sfw
2      tpe

所以我希望结果如下：

id     num
2      5

解决方法:

您可以在boolean indexing或query中使用SeriesGroupBy.nunique：

s = df.groupby('id')['airport'].nunique()
print (s)
id
1    2
2    5
3    1
Name: airport, dtype: int64

df1 = s[s > 3].reset_index()
print (df1)
   id  airport
0   2        5

要么：

df1 = df.groupby('id')['airport'].nunique().reset_index().query('airport > 3')
print (df1)
   id  airport
1   2        5