argsort() Функция возвращает матрицу индексов, которую можно использовать для индексации исходного массива, чтобы результат соответствовалsort() результат.

Есть ли способ применить эти показатели? У меня есть два массива, один - массив, используемый для получения порядка сортировки, а другой - некоторые связанные данные.

Я хотел бы вычислитьassoc_data[array1.argsort()] но это, похоже, не работает.

Вот пример:

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]])

Я хотел бы применить i к z2 (или другому массиву со связанными данными), но я не уверен, как это сделать.