numpy.matrix.partition

метод

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

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

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

Параметры:

kthint или последовательность целых чисел

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

Устарело начиная с версии 1.22.0: Передача булевых значений в качестве индекса устарела.

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]) # may vary

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