Snowpark Dataframe Group By