numpy.argwhere

numpy.argwhere(a) [source]

Нахождение индексов элементов массива, отличных от нуля, сгруппированных по элементу.

Параметры:	a : array_like Входные данные.
Возвращает:	index_array : ndarray Индексы элементов, отличных от нуля. Индексы сгруппированы по элементу.

См. также

where, nonzero

Примечания

np.argwhere(a) эквивалентно np.transpose(np.nonzero(a)).

Результат argwhere не подходит для индексирования массивов. Для этой цели используйте nonzero(a).

Примеры

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])