For Loop Iterates Less Times Than I Expected In Python
Solution 1:
You are deleting the first element in every iteration of the loop by del a[0]
, so the iterator is emptied in 3 steps, because it moves to the element after the one you removed on the next iteration.
You can check the element the iterator is currently on, and the list status in the code below
a = [1, 2, 3, 4, 5, 6]
for elem in a:
print(elem)
del a[0]
print(a)
The output is
1
[2, 3, 4, 5, 6]
3
[3, 4, 5, 6]
5
[4, 5, 6]
You can think of it as a pointer pointing to the first element of the list, that pointer jumps 2 steps when you delete the first element on each iteration, and it can jump only 3 times for 6 elements.
Generally it is a bad idea to modify the same list you are iterating on.
But if you really want to, you can iterate over the copy of the list a[:]
if you really want to delete items
a = [1, 2, 3, 4, 5, 6]
for elem in a[:]:
del a[0]
print(a)
The output is
[2, 3, 4, 5, 6][3, 4, 5, 6][4, 5, 6][5, 6][6][]
Solution 2:
The list
iterator in CPython works by iterating over the positions of the list. You can think of it working like this:
deflist_iter(items: list):
index = 0whileTrue:
yield items[index]
index += 1
In other words, iteration provides the item at 0, then 1, then 2, and so on. There is no pre-fetching of items - an item is looked up from the list when needed.
As you delete the first item on every step, the list is shortened by 1 on each step. Since you start with a list of 6 items, on the third iteration it is down to 3 items - meaning the fourth iteration fails to look up an item. Thus, your iteration finishes after three steps.
You can see this when printing also the current element in each loop. To visualise the effect, use enumerate
to get the index of the iteration. Notice that it advances by one index, but the values are also shifted for a total offset of two:
>>>a = [1, 2, 3, 4, 5, 6]...for idx, elem inenumerate(a):...print(elem, 'from', a)...print(' ', ' '*idx, '^')...del a[0]...
1 from [1, 2, 3, 4, 5, 6]
^
3 from [2, 3, 4, 5, 6]
^
5 from [3, 4, 5, 6]
^
It is generally not well-defined to modify a container while iterating over it. You should iterate over a copy instead:
a = [1, 2, 3, 4, 5, 6]
for elem in a.copy():
del a[0]
print(a)
Solution 3:
Iterating over and deleting elements from a list at the same time is tricky. One way to manage it is to traverse the list in reverse:
a = [1, 2, 3, 4, 5, 6]
for elem inreversed(a):
print(a)
del a[0]
print(a)
It'll print:
[1, 2, 3, 4, 5, 6][2, 3, 4, 5, 6][3, 4, 5, 6][4, 5, 6][5, 6][6][]
Post a Comment for "For Loop Iterates Less Times Than I Expected In Python"