Non-elementary group-by aggregations in Polars vs pandas