numpy.argwhere

numpy.argwhere(a) [source]

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

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

См. также

where, nonzero

Примечания

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

Результат argwhere не подходит для индексирования массивов. Для этой цели используйте where(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]])