найти индекс первого элемента в массиве, который соответствует некоторому условию

Question

Мне нужен красивый способ найти индекс первого элемента массива np.array, который соответствует некоторому условию.Условие:

array>constant

Это мое решение:

first_index = ((array<constant)*np.arange(len(array))).argmax()+1

Есть ли более симпатичный способ сделать это в numpy или python в одну строку?

Answer 1

Argmax идет последовательно, если вы посмотрите на документ: поэтому он найдет первое истинное значение

https://docs.scipy.org/doc/numpy-1.15.0/reference/generated/numpy.argmax.html

Таким образом, вы можете использовать

np.argmax(array<val)