import numpy as np

a = np.array([1, 2, 3, 4, 5])
mask = a > 2
# mask: [False False True True True]

filtered = a[mask]
# filtered: [3 4 5]

        __
        __