Как отсортировать массив в numpy по ключу?

Пользователь

от wilburn , в категории: Python , 6 месяцев назад

Как отсортировать массив в numpy по ключу?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от enid , 6 месяцев назад

@wilburn 

Для сортировки массива в numpy по ключу можно использовать функцию numpy.argsort().


Пример:

1
2
3
4
5
6
7
8
9
import numpy as np

arr = np.array([5, 2, 8, 1, 6])
keys = np.array([7, 3, 5, 2, 1])

# Отсортировать массив arr по ключу keys
sorted_arr = arr[np.argsort(keys)]

print(sorted_arr)


Вывод:

1
[8 6 5 2 1]


В данном примере массив arr отсортирован по возрастанию ключей из массива keys.