Python - for문에서 리스트 순회 중 요소 값 제거

pythonlist

기본적으로 아래와 같은 코드로 리스트를 순회하면서 요소를 제거하면, Index가 바뀌면서 어떤 요소는 건너띄고 순회하는 문제 등이 발생할 수 있습니다.

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

for x in numbers:
    numbers.remove(x)

위와 같은 문제를 발생시키지 않으면서, 반복문에서 리스트를 순회하면서 요소 값을 안전하게 제거하는 방법에 대해서 알아보겠습니다.

1. 리스트 복사 후, 순회하기

리스트를 먼저 복사하고, 복사된 리스트를 순회하면서 원본 리스트에서 요소를 삭제하도록 구현하면, 안전하게 리스트에서 요소 값을 삭제할 수 있습니다.

아래는 slice를 이용하여 리스트를 복사하는 예제입니다.

numbers = [1, 2, 3, 4, 5, 6]

for num in numbers[:]:  # Creating a copy of the list to iterate
    if num % 2 == 0:
        numbers.remove(num)
print(numbers)

Output:

[1, 3, 5]

slice 대신에 아래와 같이 list()로 리스트 객체를 복사할 수 있습니다.

numbers = [1, 2, 3, 4, 5, 6]

for num in list(numbers):  # Creating a copy of the list to iterate
    if num % 2 == 0:
        numbers.remove(num)
print(numbers)

Output:

[1, 3, 5]

2. 리스트 마지막에서 Index 0으로 순회

리스트를 역순으로 순회하면, 위에서 말한 Index가 바뀌면서 어떤 요소를 건너띄는 문제들이 발생하지 않습니다. 이미 순회한 Index에 대한 요소를 제거하는 것이고, 앞으로 for문에서 순회할 요소에 영향을 주지 않기 때문입니다.

아래와 같이 range()를 사용하여 마지막 Index에서 Index 0으로 순회할 수 있습니다.

numbers = [1, 2, 3, 4, 5, 6]

for i in range(len(numbers) - 1, -1, -1):
    if numbers[i] % 2 == 0:
        numbers.pop(i)
print(numbers)

Output:

[1, 3, 5]

range() 대신에 reversed()를 이용하여 역순으로 순회하도록 구현할 수도 있습니다.

numbers = [1, 2, 3, 4, 5, 6]

for i in reversed(range(len(numbers))):
    if numbers[i] % 2 == 0:
        numbers.pop(i)
print(numbers)

Output:

[1, 3, 5]
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha