Usando mientras que en la comprensión de lista o generador de expresiones

Question

puedo usar if y for en las listas por comprensión/generador de expresiones como

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

Sé que esto no es el más eficiente, pero toleradme como la condición podría ser mucho más complicado y esto es solo un ejemplo. Sin embargo, esto todavía pasa por cien iteraciones y sólo se obtiene un valor en la primera 6. ¿Hay una manera de decir la expresión generadora dónde parar con algo como esto:

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

Sin embargo, while no se entiende en el generador expresiones. Entonces, mi pregunta es, ¿cómo escribo una expresión de generador con una condición de parada para que no continúe el cálculo, incluso si no arroja nuevos valores?

Answer 1

Las diversas funciones en itertools (takewhile() me vienen a la mente) pueden ayudar.

Answer 2

Debido a que la sintaxis de takewhile() y dropwhile() no es el más claro , aquí están los ejemplos reales de su pregunta:

>>> [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] 

saber que el autor de itertools has questioned si se debe despreciar estas funciones.