Tengo una lista de comprensión que llama a una función que potencialmente devuelve None .
>>> f = lambda x: x if x < 3 else None >>> l = [f(x) for x in [1,2,3,4]] [1, 2, None, None]Me gustaría tener la lista de comprensión como la anterior, sin las entradas 'Ninguno'.
¿Cuál es una forma más eficiente de hacer lo siguiente, sin crear una sobrecarga adicional y manteniendo la eficiencia de la comprensión de la lista?
>>> filter(None, [f(x) for x in [1,2,3,4]]) [1, 2]Agregue un if en su comprensión como:
l = [y for y in (f(x) for x in [1,2,3,4]) if y is not None]Al colocar una expresión generadora dentro de la comprensión de la lista, solo necesitará evaluar la función una vez. Además, la expresión del generador es un generador, por lo que no requiere almacenamiento intermedio adicional.
A partir de Python 3.8, puede usar una Expresión de asignación (:=) (también conocida como Expresiones con nombre o el Operador Walrus) para evitar múltiples evaluaciones de f() como:
l = [y for x in [1,2,3,4] if (y := f(x)) is not None]