Comprensión de la lista de Python - simple

Question

Tengo una lista y quiero usar una determinada función solo en aquellas entradas que cumplen una cierta condición, dejando las otras entradas sin modificar.

Ejemplo: Supongamos que quiero multiplicar por 2 solo aquellos elementos que son pares.

a_list = [1, 2, 3, 4, 5] 

resultado deseado:

a_list => [1, 4, 3, 8, 5] 

Pero [elem * 2 for elem in a_list if elem %2 == 0] rendimientos [4, 8] (que actúan como un filtro, además).

¿Cuál es la forma correcta de hacerlo?

Answer 1

Utilice un conditional expression:

[x * 2 if x % 2 == 0 else x 
for x in a_list] 

(Nota del friki de la matemáticas: también se puede resolver este caso en particular con

[x * (2 - x % 2) for x in a_list] 

pero prefiero la primera opción de todos modos;)

Answer 2

a_list = [1, 2, 3, 4, 5] 

print [elem*2 if elem%2==0 else elem for elem in a_list ] 

o, si tiene una lista muy larga que desea modificar en su lugar:

a_list = [1, 2, 3, 4, 5] 

for i,elem in enumerate(a_list): 
    if elem%2==0: 
     a_list[i] = elem*2 

así, sólo las incluso elementos se modifican

Answer 3

Usted podría utilizar lambda:

>>> a_list = [1, 2, 3, 4, 5] 
>>> f = lambda x: x%2 and x or x*2 
>>> a_list = [f(i) for i in a_list] 
>>> a_list 
[1, 4, 3, 8, 5] 

Editar - Pensando en observación agf 's hice una segunda versión de mi código:

>>> a_list = [1, 2, 3, 4, 5] 
>>> f = lambda x: x if x%2 else x*2 
>>> a_list = [f(i) for i in a_list] 
>>> a_list 
[1, 4, 3, 8, 5]