numpy.flatnonzero

numpy.flatnonzero(a)

返回在 a 的展平版本中非零的索引.

这等价于 np.nonzero(np.ravel(a))[0].

参数:

aarray_like

输入数据.

返回:

resndarray

输出数组,包含 a.ravel() 元素中非零元素的索引.

参见

nonzero

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

ravel

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

示例

>>> import numpy as np
>>> 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])