Как найти n-й минимум массива - PullRequest
0 голосов
/ 20 февраля 2019

Я хочу найти индексы в порядке возрастания элементов в последней строке в двумерном массиве.Пример как ниже.

ti = np.array([[255, 2, 255, 9], [4, 255, 6, 255], [23, 11, 5, 18]])

# output = function(ti[-1])
# output : 2, 1, 3, 0

Как мне этого добиться?np.argmin дает только один минимум, и я не мог найти другого способа, включая использование раздела (который вместо этого возвращает отсортированные элементы).Пожалуйста, помогите.

1 Ответ

0 голосов
/ 20 февраля 2019

Я попробовал np.argsort, который дает такой вывод.Другие, пожалуйста, просвещайте, если еще какие-то идеи будут добавлены.

np.argsort(ti[-1])

вывод:

array([2, 1, 3, 0], dtype=int64)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...