Es más sobre la syntax de comprensión de listas de python. Tengo una lista de comprensión que produce una lista de números impares de un rango dado:
[x for x in range(1, 10) if x % 2]
Esto hace un filtro: tengo una lista de fonts, donde elimino números pares ( if x % 2
). Me gustaría usar algo como if-then-else aquí. El siguiente código falla:
>>> [x for x in range(1, 10) if x % 2 else x * 100] File "", line 1 [x for x in range(1, 10) if x % 2 else x * 100] ^ SyntaxError: invalid syntax
Hay una expresión de python como if-else:
1 if 0 is 0 else 3
¿Cómo usarlo dentro de una lista de comprensión?
x if y else z
es la syntax de la expresión que está devolviendo para cada elemento. Por lo tanto usted necesita:
[ x if x%2 else x*100 for x in range(1, 10) ]
La confusión surge del hecho de que está utilizando un filtro en el primer ejemplo, pero no en el segundo. En el segundo ejemplo, solo está asignando cada valor a otro, utilizando una expresión de operador ternario.
Con un filtro, necesitas:
[ EXP for x in seq if COND ]
Sin filtro necesitas:
[ EXP for x in seq ]
y en su segundo ejemplo, la expresión es “compleja”, que implica un if-else
.
[x if x % 2 else x * 100 for x in range(1, 10) ]
Solo otra solución, espero que a alguien le guste:
Usando: [Falso, Verdadero] [Expresión]
>>> map(lambda x: [x*100, x][x % 2 != 0], range(1,10)) [1, 200, 3, 400, 5, 600, 7, 800, 9] >>>
Puedes hacerlo con la lista de comprensión también:
A=[[x*100, x][x % 2 != 0] for x in range(1,11)] print A
Pude hacer esto
>>> [x if x % 2 != 0 else x * 100 for x in range(1,10)] [1, 200, 3, 400, 5, 600, 7, 800, 9] >>>