numpy.flatnonzero
numpy.flatnonzero(a) [source]
在a的展平版本中返回非零的索引。
这等效于np.nonzero(np.ravel(a))[0]。
参数 ： | a ：array_like 输入数据。 |
返回值 ： | res ：ndarray 输出数组，包含非零的a.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])