numpy.matrix.partition

метод

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