numpy.ma.masked_array.partition

метод

masked_array.partition(kth, axis=-1, kind='introselect', order=None) [source]

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

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

Параметры

kthint or sequence of ints

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

axisint, optional

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

kind{‘introselect’}, optional

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

orderstr or list of str, optional

Когда 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])