numpy.partition
numpy.partition(a, kth, axis=-1, kind='introselect', order=None) [source]
返回数组的分区副本。
创建一个数组的副本，并对数组中的元素进行重新排列，使第k个位置上的元素的值在已排序的数组中所处的位置。所有小于第k个元素的元素都被移到这个元素前面，所有等于或大于这个元素的元素都被移到它后面。这两个分区中元素的顺序是未定义的。
1.8.0版中的新功能。
参数 ： | a ：array_like 要排序的数组。 kth ：int 或 int的sequence 分区依据的元素索引。 元素的第k个值将处于其最终排序位置， 所有较小的元素将在其之前移动， 而所有相等或较大的元素将在其后面移动。 分区中所有元素的顺序是不确定的。 如果提供了第k个序列， 它将立即将所有按k个索引的元素划分到其排序位置。 axis ： 要排序的轴。 如果为 默认值为 kind ： 选择算法。 默认值为 或der ：str 或 str的list, 可选 当a是一个定义了字段的数组时， 这个参数指定了第一个、第二个比较的字段， 等等。可以将单个字段指定为字符串。 并非所有字段都需要指定， 但仍将使用未指定字段， 按照它们在dtype中出现的顺序，以断开连接。 |
返回值 ： | partitioned_array ：ndarray 与a具有相同类型和形状的数组。 |
Notes
各种选择算法的特征在于它们的平均速度，最差性能，工作空间大小以及它们是否稳定。 稳定的排序使具有相同键的项以相同的相对顺序保持。 可用算法具有以下属性：
沿着除最后一个轴以外的任何一个进行分区时，所有分区算法都会临时复制数据。 因此，与沿任何其他轴进行分区相比，沿最后一个轴进行分区速度更快且占用的空间更少。
复数的排序顺序是字典排序的。如果实部和虚部都是非nan，那么顺序由实部决定，除非它们相等，在这种情况下，顺序由虚部决定。
例子
>>> a = np.array([3, 4, 2, 1]) >>> np.partition(a, 3) array([2, 1, 3, 4])
>>> np.partition(a, (1, 3)) array([1, 2, 3, 4])