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])