Numpy: argsort не работает для многомерных массивов

argsort() работает нормально, но индексация работает не так, как вы ожидали.

http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#advanced -indexing

Попробуй это:

a = np.random.random([5,5])
i = np.arange(len(a))[:, np.newaxis]
j = np.argsort(a, axis=1)
a_sorted = a[i, j]
np.sort(a, axis=1)

Я запутался точно так же. В документации прямо говорится:

Возврат
index_array: ndarray, интервал
Массив индексов, сортирующих a по указанной оси.
Другими словами, a[index_array] возвращает отсортированный a .

Это применимо только к 1D-массиву. Следует ли расширить документацию, чтобы указать, что это не относится к массивам ND?

@lzkelley Да, мы очень приветствовали бы патч для улучшения документации.

@shoyer круто, сделаю пиар и

Было бы хорошо, если бы был пример того, как на самом деле использовать вывод для сортировки массива ND.

Также было бы хорошо, если бы было более читаемое решение, чем a[np.arange(np.shape(a)[0])[:,np.newaxis], np.argsort(a)]

Возможно, @seberg мог бы сделать это одним из своих дополнений к функции индексации.

Я не думаю, что он явно подходит для индексации, функция выбора или около того, вероятно, более простое соответствие.

+1 за новую функцию.

Разве вся цель argsort (в отличие от sort ) не должна использоваться для индексации?

argsort предоставляет индексы для сортировки по оси, но на самом деле сортировка по этой оси с использованием индексов кажется излишне многословной. Если нет более простого способа, о котором я не знаю.

https://github.com/numpy/numpy/blob/5becd12914d0402967205579d6f59a98151e0d98/doc/neps/indexing.rst#related -questions

функция выбора или около того, вероятно, более легкое совпадение.

Это # 8708