pandas groupby filter groups based on multiple size constraints

user_438223 0
Viewed 94

I have the following code to filter out groups with size less than 4 and greater than 51. I am issuing two statements for the same, which I believe is very inefficient. Is there a way I can say 51 > 'size' > 4

df = df[df.groupby('id',sort=False)['id'].transform('size') > 4]
df = df[df.groupby('id')['id'].transform('size') < 51]
pandas-groupby python-3.8 pandas
1 Answers

Use Series.between with inclusive=False:

df[df.groupby('id',sort=False)['id'].transform('size').between(4, 51, inclusive=False)]
user_29010020
Related