Использование в то время как в списке понимание или генератор выражений

Question

я могу использовать if и for в списковых/выражений генератора в качестве

list(i for i in range(100) if i*i < 30) 

Я знаю, что это не самый эффективный, но медведь со мной, как условие может быть гораздо больше сложный, и это всего лишь пример. Тем не менее, это все еще проходит через сотни итераций и только дает значение в первой 6. Есть ли способ сказать выражение генератора, где остановиться с чем-то вроде этого:

list(i for i in range(100) while i*i < 30) 

Однако while не понят в генераторе выражения. Итак, мой вопрос: как написать выражение генератора с условием остановки, чтобы он не продолжал вычисление, даже если он не дает новых значений.

Answer 1

Различные функции в itertools (takewhile() приходят на ум) могут помочь.

Answer 2

Поскольку синтаксис takewhile() и dropwhile() не ярким, вот реальные примеры вашего вопроса:

>>> [i for i in itertools.takewhile(lambda x: x*x<30, range(10))] 
[0, 1, 2, 3, 4, 5] 
>>> [i for i in itertools.dropwhile(lambda x: x*x<30, range(10))] 
[6, 7, 8, 9] 

Знайте, что автор itertools has questioned ли принизить эти функции.