Problem z usuwaniem elementów z listy w Python

Question

Cześć,

przy testowaniu pętli for i usuwaniu z jej pomocą liczb parzystych z listy napotkałem na problem, który nie bardzo wiem jak rozwiązać.

thislist = [1,2,2,3,4]

for n in thislist:
    if n % 2 == 0:
        thislist.remove(n)
        print(n)
        print(thislist)

Wynik:

2
[1, 2, 3, 4]
4
[1, 2, 3]

Problem polega na tym, że usuwa liczby 2 i 4 z pominięciem jednej "dwójki". Natomiast samo działanie pętli for bez usuwania z listy wygląda tak:

thislist = [1,2,2,3,4]

for n in thislist:
    if n % 2 == 0:
        print(n)

Wynik:

2
2
4

Answer 1

Ogólnie modyfikowanie listy podczas iteracji to jest zazwyczaj zły pomysł. W Twoim przypadku remove() powoduje, że iteracja "przeskoczy" jeden element.

Działającym ale brzydkim rozwiązaniem jest iterowanie po kopii listy (wtedy nie jest modyfikowana):

for n in list(thislist):
    if n % 2 == 0:
        thislist.remove(n)

Ale znacznie lepszym rozwiązaniem, w tym algorytmicznie, jest użycie list comprehension:

thislist = [n for n in thislist if n % 2 != 0]

 

Comments

Dzięki wielkie Adrian, zwłaszcza za tę drugą opcję "list comprehension", której nie znałem, ale nazwa mi się obiła o uszy. Chyba bardziej zainteresuję się tym tematem

---

Zawsze też można iterować od końca:

thislist = [1,2,2,3,4]

for n in reversed(thislist):
    if n % 2 == 0:
        thislist.remove(n)
        print(n)
        print(thislist)