https://tanvirislam3113.blogspot.com/2023/04/pandas-dataframe-selecting-groups-with.html