排序、搜索和计数#

排序#

sort(a[, axis, kind, order, stable])

返回数组的已排序副本。

lexsort(keys[, axis])

使用一系列键执行间接稳定排序。

argsort(a[, axis, kind, order, stable])

返回将对数组进行排序的索引。

ndarray.sort([axis, kind, order])

对数组进行就地排序。

sort_complex(a)

使用实部优先,然后虚部优先对复数数组进行排序。

partition(a, kth[, axis, kind, order])

返回数组的已分区副本。

argpartition(a, kth[, axis, kind, order])

使用 kind 关键字指定的算法,在给定轴上执行间接分区。

搜索#

argmax(a[, axis, out, keepdims])

返回沿轴的最大值的索引。

nanargmax(a[, axis, out, keepdims])

返回指定轴上最大值的索引,忽略 NaN。

argmin(a[, axis, out, keepdims])

返回沿轴的最小值的索引。

nanargmin(a[, axis, out, keepdims])

返回指定轴上最小值的索引,忽略 NaN。

argwhere(a)

找到非零数组元素的索引,按元素分组。

nonzero(a)

返回非零元素的索引。

flatnonzero(a)

返回 a 的扁平化版本中非零的索引。

where(condition, [x, y], /)

根据 condition 返回从 xy 中选择的元素。

searchsorted(a, v[, side, sorter])

找到应插入元素以保持顺序的索引。

extract(condition, arr)

返回满足某些条件的数组元素。

计数#

count_nonzero(a[, axis, keepdims])

计算数组 a 中非零值的个数。