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