https://camera-optical-zoom.blogspot.com/2023/01/python-groupby-with-boolean-mask.html