[Python List] for loop 안에서 list.remove() 사용 시 주의할 점

문제상황

---

파이썬 리스트를 for loop로 순회하면서 아이템의 값이 3 미만인 경우 리스트에서 삭제하고자 합니다.

mylist엔 1, 2 값이 삭제된 [3, 4, 5]가 들어있을 것이라 예상하였는데,

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

for item in mylist:
    if item < 3:
        mylist.remove(item)

 

결과는 아래 처럼 1만 삭제되었습니다.

print(mylist)
# [2, 3, 4, 5] 

 

2는 왜 삭제 되지 않았을까요?

디버깅해보면

• item이 0번 인덱스를 참조하여 mylist.remove(1)이 수행되어 1이 삭제되고

• 다음인덱스인 1번 인덱스를 참조했을 때, 변경된 리스트에서 1번 인덱스의 값인(3)을 참조하게 되면서
  삭제 전 1번 인덱스였던 2를 건너뛰게 됩니다.

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

for item in mylist:
    if item < 3:
				print(mylist,item)
        mylist.remove(item)


# [1, 2, 3, 4, 5] 1 (item-> index 0: 1)
# [2, 3, 4, 5] 3    (item-> index 1: 3)
# [2, 3, 4, 5] 4    (item-> index 2: 4)
# [2, 3, 4, 5] 5    (item-> index 2: 5)

 

 

해결방법

---

원본 리스트(mylist)가 아닌 리스트의 copy 한 값(mylist[:])을 순회합니다.

순회할 대상이 되는 copy list의 값을 직접적으로 변경하지 않기 때문에 모든 item들을 검사 할 수 있습니다.

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

for item in mylist[:]:
    if item < 3:
        mylist.remove(item)

print(mylist)
# [3, 4, 5]

 

 

Pythonic approach

---

위의 코드는 아래와 같이 한줄로 작성가능합니다.

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

mynewlist = [item for item in mylist if item >= 3]

print(mynewlist)
[3, 4, 5]

 

 

Link

---

https://gist.github.com/alexlouden/9f1ab4354d1c68ae4c1c94126ac51a20

Iterating while modifying your list.md