https://llego.dev/posts/numpy-aggregations-python/