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