numpy.ndarray.partition

method

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

以某种方式对数组中的元素进行部分排序,使第 k 个位置的元素的值位于排序数组中应处的的位置.在输出数组中,所有小于第 k 个元素的元素都位于该元素的左侧,所有等于或大于该元素的元素都位于其右侧.输出数组中第 k 个元素两侧的两个分区中元素的顺序未定义.

参数:

kthint 或 int 序列

用于分区的元素索引.第 k 个元素值将位于其最终排序位置,并且所有较小的元素将移到它之前,并且所有相等或更大的元素都将位于它之后.分区中所有元素的顺序未定义.如果提供了第 k 个元素的序列,它将立即将所有由第 k 个元素索引的元素划分为它们排序后的位置.

自 1.22.0 版本弃用: 不建议将布尔值作为索引传递.

轴int, optional

沿其排序的轴. 默认为 -1,表示沿最后一个轴排序.

kind{‘introselect’}, 可选

选择算法.默认为"introselect".

orderstr 或 list of str,可选

当 a 是定义了字段的数组时,此参数指定首先比较哪些字段,然后比较第二个等等.单个字段可以指定为字符串,并且不需要指定所有字段,但仍将使用未指定的字段,按照它们在 dtype 中出现的顺序来打破联系.

参见

numpy.partition

返回数组的分区副本.

argpartition

间接划分.

sort

完全排序.

注释

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

示例

Try it in your browser!

>>> 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])

Go BackOpen In Tab