Python: comprensión de la lista para asignar diferentes valores

Question

Estoy haciendo una lista en 2D y me gustaría inicializarla con una lista de comprensión. Me gustaría que haga algo como esto:

[[x for i in range(3) if j <= 1: x=1 else x=2] for j in range(3)] 

por lo que debe devolver algo como:

[[1,1,1], 
[1,1,1], 
[2,2,2]] 

¿Cómo podría yo ir haciendo esto?

Gracias por su ayuda.

Answer 1

Parece como si estuviera buscando algo como esto:

[[1 if j <= 1 else 2 for i in range(3)] for j in range(3)] 

El pitón conditional expression es un poco diferente de lo que pueda ser utilizada si viene de algo como C o Java:

La expresión x if C else y evalúa primero C (no x); si C es verdadero, x se evalúa y se devuelve su valor; de lo contrario, se evalúa y y se devuelve su valor.

Una forma un poco más corto para hacer la misma cosa es:

[[1 if j <= 1 else 2]*3 for j in range(3)] 

Answer 2

respuesta de Greg es correcta, aunque una expresión mucho más simple y más rápido para producir el resultado deseado sería

[[j] * 3 for j in (1, 1, 2)] 

es decir, recuerde que for no necesita aplicarse a un range solamente ;-), existe multiplicación de listas, y así sucesivamente ;-).

Answer 3

try que

>>> [[(1 if j<1 else 2) for i in range(3)] for j in range(3)] 
[[1, 1, 1], [2, 2, 2], [2, 2, 2]] 

La segunda vez j = 1, de modo j < 1 falla