Индексы частичной сортировки в NumPy
Функция
np.argpartition не возвращает
сами значения: она выдаёт индексы,
которые указывают, откуда брать
элементы, чтобы та же перестановка
выполнялась по смыслу, что и у
np.partition.
Снова возьмём ряд оценок и позицию
2. Выведем массив индексов:
import numpy as np
scores = np.array([5, 2, 8, 1, 6])
idx = np.argpartition(scores, 2)
print(idx) # выведет [3 1 0 2 4]
Для ряда marks [6, 1, 4, 2, 5]
получите индексы частичной
сортировки с позицией 2
и выведите их.
Для ряда prices [100, 50, 80, 20]
найдите индексы, соответствующие
частичной сортировке с позицией
1, и выведите результат.
