numpy.argwhere#

numpy.argwhere(a)[источник]#

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

Параметры:

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

Возвращает:

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

Смотрите также

where, nonzero

Примечания

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

Вывод argwhere не подходит для индексации массивов. Для этой цели используйте nonzero(a) вместо этого.

Примеры

>>> import numpy as np
>>> 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]])