Basic Python Dataframe Group By Count In Python For Loop