Булева индексация

Сравнение даёт массив bool, которым можно отфильтровать элементы. Удобно для условной выборки и присваивания.

import numpy as np
a = np.array([1, 5, 2, 8, 3])
print(a[a > 3])   # [5 8]
a[a > 3] = 0
print(a)          # [1 0 2 0 3]