https://llego.dev/posts/pandas-filtering-groups/