https://llego.dev/posts/python-pandas-split-apply-combine-groupby/