numpy.ndarray.partition

метод

ndarray.partition(kth, axis=- 1, kind='introselect', order=None)

Переупорядочивает элементы в массиве таким образом, что значение элемента в позиции kth находится в той позиции, в которой оно должно быть в отсортированном массиве. Все элементы, меньшие, чем элемент в позиции kth, перемещаются перед ним, а все равные или большие - после него. Порядок элементов в двух разделах не определен.

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

Параметры

kthint или последовательность индексов

Индекс элемента для разбиения. Значение элемента в позиции kth будет находиться в его конечном отсортированном положении, и все меньшие элементы будут перемещены перед ним, а все равные или большие элементы - после него. Порядок всех элементов в разделах не определен. Если указана последовательность kth, все элементы, индексированные значениями из kth, будут разбиениы в их отсортированное положение одновременно.

axisint, необязательно

Ось, по которой необходимо выполнить сортировку. По умолчанию -1, что означает сортировку по последней оси.

kind{‘introselect’}, необязательно

Алгоритм выбора. По умолчанию ‘introselect’.

orderстрока или список строк, необязательно

Когда a является массивом с определенными полями, этот аргумент определяет, какие поля сравнивать в первую, вторую и т.д. очередь. Одно поле может быть указано в виде строки, и не все поля нужно указывать, но не указанные поля все равно будут использоваться в порядке их появления в dtype, чтобы разрешить ничьи.

См. также

numpy.partition

Возвращает отсортированную копию массива.

argpartition

Непрямое разбиение.

sort

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

Примечания

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

Примеры

>>> a = np.array([3, 4, 2, 1])
>>> a.partition(3)
>>> a
array([2, 1, 3, 4])

>>> a.partition((1, 3))
>>> a
array([1, 2, 3, 4])