https://llego.dev/posts/boolean-arrays-numpy/