Индексы подходящих элементов в NumPy

Функция nonzero принимает логический массив или условие над данными и возвращает индексы ячеек, где значение истинно. Для одномерного ряда результат это кортеж с одним массивом номеров.

Сравним ряд с порогом и выведем индексы, где условие выполнилось:

import numpy as np

scores = np.array([1, 4, 3, 5])
print(np.nonzero(scores > 3)) # выведет (array([1, 3]),)

Номера идут в порядке возрастания. Для другого условия список индексов будет другим:

import numpy as np

temps = np.array([10, 20, 15, 25])
print(np.nonzero(temps >= 18)) # выведет (array([1, 3]),)

Создайте ряд из чисел 2, 7, 5 и 9 и выведите индексы ячеек, где значение больше 6.

Задайте ряд из пяти целых чисел, найдите номера ячеек, равных 0, и выведите эти номера.

Создайте ряд из чисел 1, 2, 2 и 3 и выведите индексы, где значение совпадает с 2.