https://llego.dev/posts/boolean-indexing-masking-in-numpy/