That’s why I said argsort :)

https://numpy.org/doc/stable/reference/generated/numpy.argso...

I assume this name derives from argmin and argmax, which go back at least to MATLAB. But MATLAB has sort returning indices too.