Databricks Python Count Rows In Dataframe Groupby