как найти индекс элемента numpy

Самым простым способом будем использование метода .where()

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 2])
indices = np.where(arr == 2)
print(indices[0])  # [1 5]

Как видно выше, метод выведет все совпадения с элементом.