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 или список 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])