Как отсортировать массив в numpy по ключу?
1 ответ
от
enid ,
год назад
@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.
15