numpy.matrix.partition

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

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

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

Параметры:

kth : int или последовательность int Индекс элемента для разбиения. Значение k-го элемента будет находиться в своей конечной отсортированной позиции, и все меньшие элементы будут перемещены перед ним, а все равные или большие — за ним. Порядок всех элементов в разделах не определен. Если указана последовательность kth, все элементы, индексированные kth, будут разделены в их отсортированную позицию сразу. axis : int, необязательно Ось, вдоль которой выполнять сортировку. По умолчанию -1, что означает сортировку вдоль последней оси. kind : {‘introselect’}, необязательно Алгоритм выбора. По умолчанию «introselect». order : str или список str, необязательно Когда a представляет собой массив с определенными полями, этот аргумент указывает, какие поля сравнивать первыми, вторыми и т. д. Одно поле можно указать как строку, и не все поля необходимо указывать, но не указанные поля все равно будут использованы в порядке их появления в dtype, чтобы разбить связи.

См. также

numpy.partition

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

argpartition

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

sort

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

Примечания

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

Примеры

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

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