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

НазадОткрыть во вкладке