https://mingredbestcheapprice.blogspot.com/2023/11/pandas-groupby-custom-groups.html