numpy.argpartition

numpy.argpartition(a, kth, axis=-1, kind='introselect', order=None) [source]

Выполняет косвенное разбиение по заданной оси с использованием алгоритма, указанного ключевым словом kind. Возвращает массив индексов той же формы, что и a, которые индексируют данные вдоль заданной оси в упорядоченном по разбиению порядке.

Введено в версии 1.8.0.

Параметры:

a : array_like Массив для сортировки. kth : int или последовательность целых чисел Индекс элемента для разбиения. Элемент с индексом kth будет находиться на своём конечном упорядоченном месте, все меньшие элементы будут перемещены перед ним, а все большие — после него. Порядок всех элементов в разбиениях не определён. Если передана последовательность значений kth, то все они будут разбросаны по своим упорядоченным местам сразу. axis : int или None, необязательно Ось, по которой выполнить сортировку. По умолчанию -1 (последняя ось). Если None, используется сплюснутый массив. kind : {‘introselect’}, необязательно Алгоритм выбора. По умолчанию ‘introselect’ order : str или список строк, необязательно Когда a является массивом с определёнными полями, этот аргумент указывает, какие поля сравнивать сначала, второе и т. д. Одно поле может быть указано как строка, и не все поля могут быть указаны, но не указанные поля всё равно будут использованы в порядке их появления в dtype, чтобы разрешать ничьи.

Возвращает:

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

См. также

partition

Описание используемых алгоритмов разбиения.

ndarray.partition

Разбиение на месте.

argsort

Полная косвенная сортировка

Примечания

См. partition для заметок об различных алгоритмах выбора.

Примеры

Одномерный массив:

>>> x = np.array([3, 4, 2, 1])
>>> x[np.argpartition(x, 3)]
array([2, 1, 3, 4])
>>> x[np.argpartition(x, (1, 3))]
array([1, 2, 3, 4])

>>> x = [3, 4, 2, 1]
>>> np.array(x)[np.argpartition(x, 3)]
array([2, 1, 3, 4])