Zrozumienie listy Pythona - proste

Question

Mam listę i chcę użyć pewnej funkcji tylko na tych wpisach, które spełniają pewien warunek - pozostawiając pozostałe wpisy niezmodyfikowane.

Przykład: Załóżmy, że pomnożę przez 2 tylko te elementy, które są parzyste.

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

kupna wynik:

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

Ale [elem * 2 for elem in a_list if elem %2 == 0] plony [4, 8] (działał jako filtr dodatkowo).

Jaki jest prawidłowy sposób na rozwiązanie tego problemu?

Answer 1

Użyj conditional expression:

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

(uwaga Math Geek jest: można również rozwiązać ten konkretny przypadek z

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

ale wolałbym pierwszą opcję i tak;)

Answer 2

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

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

lub, jeśli masz bardzo długą listę, którą chcesz zmodyfikować w miejscu:

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

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

tak, tylko nawet elementy są modyfikowane

Answer 3

Można użyć 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] 

Edit - Myślenie o uwaga agf „s Zrobiłem 2nd wersję mojego kodu:

>>> 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]