losargsort() La función devuelve una matriz de índices que se puede usar para indexar la matriz original de modo que el resultado coincida con elsort() resultado.

¿Hay alguna forma de aplicar esos índices? Tengo dos matrices, una es la matriz utilizada para obtener el orden de clasificación y otra es algunos datos asociados.

Me gustaría computarassoc_data[array1.argsort()] Pero eso no parece funcionar.

Aquí hay un ejemplo:

z=array([1,2,3,4,5,6,7])
z2=array([z,z*z-7])
i=z2.argsort()

z2=array([[ 1,  2,  3,  4,  5,  6,  7],
          [-6, -3,  2,  9, 18, 29, 42]])
i =array([[1, 1, 1, 0, 0, 0, 0],
          [0, 0, 0, 1, 1, 1, 1]])

Me gustaría aplicar i a z2 (u otra matriz con datos asociados) pero no estoy seguro de cómo hacerlo.