Funkcja remove() nie usuwa elementów

Question

Hej, ostatnio wyhaczyłem taką zależność podczas używania funkcji remove():

list = [2.0, 2.0, 1.0, 1.45, 2.0]

for el in list:
    if el == 2.0:
        list.remove(el)

print(list)

Output:
[1.0, 1.45, 2.0]

Natomiast przy wejściu gdzie nie ma podciągu elementów usuwanych, usuwa wszystkie:

list = [2.0, 1.0, 1.45, 2.0]

for el in list:
    if el == 2.0:
        list.remove(el)

print(list)

Output:
[1.0, 1.45]

Ktoś wie dlaczego tak się dzieje? Dosyć zastanawiające zjawisko mi się wydaje. Na Internecie nie mogłem znaleźć podobnego pytania.

Answer 1

Chodzi o to, że mutujesz listę podczas jej iterowania, co zawsze prowadzi do nieintuicyjnego zachowania - dlatego powinno się tego unikać.

Konkretniej: remove() usuwa pierwsze wystąpienie argumentu w liście. Jak dodasz print()a na początku...

for el in list:
    print(el)
    if el == 2.0:
        print("usuwam!")
        list.remove(el)

To zobaczysz coś takiego:

# dla [2.0, 2.0, 1.0, 1.45, 2.0]
2.0
usuwam!
1.0
1.45
2.0
usuwam!

# dla [2.0, 1.0, 1.45, 2.0]
2.0
usuwam!
1.45
2.0
usuwam!

Zauważ, że po usunięciu, następna iteracja przeskakuje jeden element - jest tak, bo przez usunięcie indeksy przesunęło się w lewo. Dlatego w pierwszym przypadku przeskakujesz jedno 2.0, więc remove() wywołuje się 2 razy mimo że są 3 elementy.

Natomiast ogólnie to co napisałeś jest... dziwne ;) Bo iterujesz się po liście i sprawdzasz czy ma elementy, ale remove() też się iteruje po liście żeby znaleźć pierwszy element do usunięcia. Więc wychodzi z tego takie przekombinowane masło maślane o kwadratowej złożoności obliczeniowej.

Zamiast tego, można prościej zapisać (zmienilem nazwe zmiennej na `tab` bo `list` to wbudowana funkcja):

while 2.0 in tab:
    tab.remove(2.0)

Ale jeszcze lepiej (unika podwójnej iteracji po liście) jest użyć składni list comprehension lub ew filter():

tab = [e for e in tab if e != 2.0]

# lub:
tab = list(filter(lambda e: e != 2.0, tab))

 

Comments

Teraz jest wszystko zrozumiałe ;) Dzięki wielkie

Answer 2

Tu jest wszystko wyjaśnione:

https://sopython.com/canon/95/removing-items-from-a-list-while-iterating-over-the-list/

https://stackoverflow.com/questions/6500888/removing-from-a-list-while-iterating-over-it

Możesz nawet podejrzeć jak dokładnie wygląda lista przy każdej iteracji i który element jest obecnie sprawdzany:

l = [2.0, 2.0, 1.0, 1.45, 2.0]
 
for el in l:
    print(el, l)
    if el == 2.0:
        l.remove(el)