排序、搜索和计数

提示

NumPy API 参考：排序、搜索和计数

排序

sort(a[, axis, kind])	使用稳定排序算法返回数组的排序副本。
lexsort(keys)	使用键数组执行间接排序。
argsort(a[, axis, kind])	返回使用稳定排序对数组进行排序后的索引。
msort(a)	返回沿第一个轴排序的数组副本。
sort_complex(a)	先使用实部，然后使用虚部对复数数组进行排序。
partition(a, kth[, axis])	返回数组的分区副本。
argpartition(a, kth[, axis])	返回将对数组进行部分排序的索引。

另请参阅

cupy.ndarray.sort()

搜索

argmax(a[, axis, dtype, out, keepdims])	返回沿轴的最大值的索引。
nanargmax(a[, axis, dtype, out, keepdims])	返回指定轴中忽略 NaN 的最大值的索引。
argmin(a[, axis, dtype, out, keepdims])	返回沿轴的最小值的索引。
nanargmin(a[, axis, dtype, out, keepdims])	返回指定轴中忽略 NaN 的最小值的索引。
argwhere(a)	返回非零元素的索引。
nonzero(a)	返回非零元素的索引。
flatnonzero(a)	返回展平后的数组中非零元素的索引。
where(condition[, x, y])	根据条件返回 x 或 y 中的元素。
searchsorted(a, v[, side, sorter])	找到应插入元素以保持顺序的索引。
extract(condition, a)	返回满足某个条件的数组元素。

计数

count_nonzero(a[, axis])

计算数组中非零值的数量。