cupy.partition

cupy.partition(a, kth, axis=-1)

返回数组的一个经过划分的副本。

创建一个数组副本，其元素被重新排列，使得位于 k-th 位置的元素的值如同在排序后的数组中一样位于该位置。所有在新 k-th 元素之前的元素都小于或等于在新 k-th 元素之后的元素。

参数:

• a (cupy.ndarray) – 要划分的数组。

• kth (int 或 int 序列) – 用于划分的元素索引。如果提供 k-th 序列，它将一次性把所有由这些 k-th 索引的元素划分到其排序位置。

• axis (int 或 None) – 沿其划分的轴。默认值为 -1，表示沿最后一个轴划分。如果提供 None，则在划分之前将数组展平。

返回:

类型和形状与 a 相同的数组。

返回类型:

cupy.ndarray

另请参阅

numpy.partition()