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