Каков самый pythonic способ перебора списка до тех пор, пока не будет выполнено условие?

Question

У меня есть список действительных значений:

values = [0.1, 2.9, 1.4, 5.7, 9.2, 3.8] 

Я хочу, чтобы количество элементов итерации, пока некоторое условие не было выполнено. Например, если условие было item > 5, результат будет 3.

Самый простой способ, и, возможно, самый прямой способ сделать это было бы:

for index, item in enumerate(values): 
    if (item > 5): 
     print(index) 
     break 

Есть ли другой способ более pythonic, предпочтительно однострочный, что делает это?

Answer 1

Вы можете использовать itertools.takewhile с sum, который будет принимать элементы, пока условие не будет выполнено, используя выражение генератора мы лениво оценить:

values = [0.1, 2.9, 1.4, 5.7, 9.2, 3.8] 

from itertools import takewhile 

print(sum(1 for _ in takewhile(lambda x: x< 5,values))) 

Answer 2

Ваш код эквивалентен следующему выражению генератора

In [1]: gen = (i for i, v in enumerate(values) if v > 5) 

next(gen) будет возвращать индекс первого совпадающего элемента

In [2]: next(gen) 
Out[2]: 3 

или поднять StopIteration (если нет таких элементов) - если это проблема рассмотреть вопрос о добавлении некоторого значения по умолчанию, например None:

In [3]: next((i for i, v in enumerate(values) if v > 10), None) is None 
Out[3]: True 

или с помощью itertools.takewhile с sum (обратите внимание, что вам нужно отрицать предикат):

In [4]: import itertools 

In [5]: it = itertools.takewhile(lambda x: x <= 5, values) 

In [6]: sum(1 for _ in it) 
Out[6]: 3 

Answer 3

Для списков, ответ Padraic! Использование NumPy, безусловно, самый быстрый, если структура данных уже массив ...

In [370]: a = np.array(values) 

In [371]: (a>5).argmax() 
Out[371]: 3 

Answer 4

Самый вещий способ сделать это со списком понимания.

print next(ind for ind, i in enumerate([0.1, 2.9, 1.4, 5.7, 9.2, 3.8]) if i > 5)