numpy.ndarray.partition#

方法

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

以一种方式部分地对数组中的元素进行排序，使得 k-th 位置上的元素的值处于其在排序数组中的位置。在输出数组中，所有小于 k-th 元素的值都位于其左侧，所有等于或大于其的值都位于其右侧。输出数组中 k-th 元素两侧分区的元素顺序是未定义的。

参数:

kth整数或整数序列: 用于分区的元素索引。k-th 元素的值将处于其最终排序位置，所有较小的元素将被移到其前面，所有等于或较大的元素将被移到其后面。分区中所有元素的顺序是未定义的。如果提供了 k-th 的序列，它将一次性将所有由 k-th 索引的元素分区到其排序位置。

已弃用，版本 1.22.0 起：将布尔值作为索引已弃用。
axisint, optional: 沿哪个轴进行排序。默认值为 -1，表示沿最后一个轴排序。
kind{‘introselect’}, 可选: 选择算法。默认值为 ‘introselect’。
orderstr 或 str 列表, optional: 当 a 是一个具有已定义字段的数组时，此参数指定首先、其次等比较哪些字段。单个字段可以指定为字符串，并非所有字段都需要指定，但未指定的字段仍将用于打破平局，顺序与它们在 dtype 中出现的顺序相同。

另请参阅

备注

有关不同算法的说明，请参见 np.partition。

示例

>>> import numpy as np
>>> 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])