排序、搜索和计数

提示

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)	返回在 a 的展平版本中非零的索引。
where(condition[, x, y])	根据条件，从 x 或 y 返回元素。
searchsorted(a, v[, side, sorter])	查找元素应插入以保持顺序的索引。
extract(condition, a)	返回满足某些条件的数组元素。

计数

count_nonzero(a[, axis])

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