numpy.flatnonzero

numpy.flatnonzero(a)[source]

Возвращает индексы, которые отличны от нуля в сплющенной версии a.

Это эквивалентно np.nonzero(np.ravel(a))[0].

Параметры

aarray_like

Входные данные.

Возвращает

resndarray

Массив вывода, содержащий индексы элементов 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])