Cómo usar la list comprehension en Python

La list comprehension (comprensión de listas) es una construcción utilizada en Python para crear una lista a partir de una expresión generadora.

list = [expresión]

La expresión generadora se coloca entre corchetes.

La comprensión de listas le permite crear automáticamente una lista de elementos múltiples sin usar bucles ni iteraciones.

Ejemplos

Ejemplo 1

Crear una lista con cuadrados de números del 1 al 10.

>>> lista = [x**2 for x in range(1,11)]

La comprensión de listas crea una lista con los números del 1 al 10 al cuadrado.

>>> lista
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

Ejemplo 2 ( list comprehension con IF )

Crear una lista con cuadrados de números pares.

Se obtiene agregando un if al ejemplo anterior.

>>> lista = [x**2 for x in range(1,11) if x%2==0]

La condición adicional if x%2==0 selecciona números divisibles por dos con resto igual a cero (números pares).

El contenido de la lista es

>>> lista
[4, 16, 36, 64, 100]

La list comprehension selecciona y eleva al cuadrado solo números pares entre 1 y 10.