https://cerwinvegassalee.blogspot.com/2024/03/python-issue-with-groupby-and-apply.html