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