numpy.argpartition

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

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

Добавлена в версии 1.8.0.

Параметры:

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