Dapatkan indeks nilai dalam daftar jika nilai memenuhi kondisi python

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