numpy.ndarray.partition#

方法

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

部分排序数组中的元素,使得第 k 个位置的元素在其在排序数组中的位置.在输出数组中,所有小于第 k 个元素的元素位于该元素的左侧,所有等于或大于的元素位于其右侧.输出数组中第 k 个元素两侧的两个分区的元素顺序未定义.

在 1.8.0 版本加入.

参数:

kth整数或整数序列: 按其分区的元素索引.第k个元素值将处于其最终排序位置,所有较小的元素将移到它前面,所有相等或更大的元素将移到它后面.分区中所有元素的顺序未定义.如果提供了一个kth序列,它将一次性将所有由kth索引的元素分区到它们的排序位置.

自 1.22.0 版本弃用: 将布尔值作为索引传递已被弃用.
axisint, 可选: 要排序的轴.默认为 -1,这意味着沿最后一个轴排序.
kind{‘introselect’}, 可选: 选择算法.默认是 ‘introselect’.
orderstr 或 str 列表,可选: 当 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])