https://lacucinadiadine.blogspot.com/2024/05/pandas-groupby-custom-groups.html