[ python ] list에서 for문으로 remove() 시 주의하기 - list[:]

list에서 for문을 돌리면서 remove()할 때

reserve = [1,2,3,4,5]
lost = [1,2,3,4,5]
for k in reserve:
        if k in lost:
            reserve.remove(k)
            lost.remove(k)
print(reserve) # [2,4]
print(lost) # [2,4]

reserve 배열과 lost배열이 같으므로 remove했을 때 둘 다 빈 배열 [],[] 을 출력하기를 기대했는데 [2,4], [2,4] 와 같은 결과가 나왔다.

remove()가 원본 배열 요소를 삭제하고 이를 적용해서 for문을 돌리기 때문에 잘못된 결과가 나온다.

해결방법

이를 해결하기 위해서 복사를 사용해야한다.

reserve = [1,2,3,4,5]
lost = [1,2,3,4,5]
for k in reserve[:]:
        if k in lost[:]:
            reserve.remove(k)
            lost.remove(k)
print(reserve) # []
print(lost) # []

이와 같이 reserve[:], lost[:] 로 배열을 복사해서 for문을 돌리고 remove를 하면 복사본으로 for문을 돌리고 원본 배열에서 요소 삭제를 하게 되기 떄문에 원하는 결과인 빈 배열을 얻을 수 있다.

주의

복사를 한다고

reserveArr = reserve
lostArr = lost
for k in reserve:
     if k in lost:
        reserveArr.remove(k)
        lostArr.remove(k)
print(reserveArr) # [2,4] 
print(lostArr) # [2,4]

 이렇게 복사를 한다면 reserveArr 과 reserve가 바라보는 주소는 동일하기 때문에 첫번째 코드와 같은 결과가 나온다..!! 복사 시 주의!