numpy.flatnonzero

numpy.flatnonzero(a)

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

这相当于np.nonzero（np.ravel（a））。 [0] .

参数

aarray_like

输入数据。

返回

res恩达雷

输出数组，包含 a.ravel() 这不是零。

参见

nonzero

返回输入数组的非零元素的索引。

ravel

返回包含输入数组元素的一维数组。

实例

>>> x = np.arange(-2, 3)
>>> x
array([-2, -1,  0,  1,  2])
>>> np.flatnonzero(x)
array([0, 1, 3, 4])

使用非零元素的索引作为索引数组来提取这些元素：

>>> x.ravel()[np.flatnonzero(x)]
array([-2, -1,  1,  2])

numpy.argwhere numpy.searchsorted