Obter índice de valor na lista se o valor atender à condição python

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