Получить индекс значения в списке, если значение соответствует условию Python

>>> import numpy as np
>>> a = np.array([1, 2, 3, 1, 2, 3])
>>> np.where(a > 2)[0]
[2 5]
Tirbo06