numpy.argwhere

numpy.argwhere(a) [source]

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

Параметры

aarray_like

Входные данные.

Возвращаемые значения

index_array(N, a.ndim) ndarray

Индексы элементов, отличных от нуля. Индексы сгруппированы по элементу. Этот массив будет иметь форму (N, a.ndim), где N — количество ненулевых элементов.

См. также

where, nonzero

Примечания

np.argwhere(a) почти такое же, как np.transpose(np.nonzero(a)), но генерирует результат правильной формы для массива размерности 0.

Результат 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]])