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 является массивом с определёнными полями, этот аргумент указывает, какие поля сравнивать в первую очередь, во вторую и т. д. Одно поле может быть указано как строка, и не все поля должны быть указаны, но не указанные поля всё равно будут использоваться в порядке их появления в типе данных, чтобы разрешать связи.

Возвращает:

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])