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’。

orderstr 或 str 列表,可选

a 是一个定义了字段的数组时,此参数指定要首先比较哪些字段、其次比较哪些字段等等。可以将单个字段指定为字符串,并且不需要指定所有字段,但是未指定的字段仍将使用,按照它们在 dtype 中出现的顺序来打破平局。

另请参阅

numpy.partition

返回数组的分区副本。

argpartition

间接分区。

sort

完整排序。

备注

有关不同算法的说明,请参阅 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])