Координаты истинных элементов в NumPy
Функция argwhere
для двумерного массива возвращает
таблицу: каждая строка это пара
(строка, столбец). Функция
nonzero при том же условии
даёт кортеж из отдельных массивов
индексов по осям; argwhere
удобнее, когда координаты нужны
одним списком пар.
Найдём координаты положительных
значений в таблице 2×2
и выведем их:
import numpy as np
flags = np.array([[0, 1], [2, 0]])
coords = np.argwhere(flags > 0)
print(coords) # выведет [[0 1] [1 0]]
В таблице [[0, 5], [5, 0], [0, 0]]
найдите координаты ячеек со
значением 5 и выведите их.
В таблице [[1, 0, 1], [0, 0, 2]]
найдите координаты элементов
больше 0 и выведите число
строк в результате.
