numpy.recarray.partition

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

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

Новая в версии 1.8.0.

Параметры:

kth : 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(3)
>>> a
array([2, 1, 3, 4])

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