Modifying list while iterating

Question:

l  = range(100)                         
for i in l:                         
    print i,                         
    print l.pop(0),                  
    print l.pop(0)

The above python code gives the output quite different from expected. I want to loop over items so that I can skip an item while looping.

Please explain.

Asked By: Xolve

||

Answers:

The general rule of thumb is that you don’t modify a collection/array/list while iterating over it.

Use a secondary list to store the items you want to act upon and execute that logic in a loop after your initial loop.

Answered By: Paul Sasik

Try this. It avoids mutating a thing you’re iterating across, which is generally a code smell.

for i in xrange(0, 100, 3):
    print i

See xrange.

Answered By: Hank Gay

I’ve been bitten before by (someone else’s) “clever” code that tries to modify a list while iterating over it. I resolved that I would never do it under any circumstance.

You can use the slice operator mylist[::3] to skip across to every third item in your list.

mylist = [i for i in range(100)]
for i in mylist[::3]:
    print(i)

Other points about my example relate to new syntax in python 3.0.

  • I use a list comprehension to define mylist because it works in Python 3.0 (see below)
  • print is a function in python 3.0

Python 3.0 range() now behaves like xrange() used to behave, except it works with values of arbitrary size. The latter no longer exists.

Answered By: Ewan Todd

Never alter the container you’re looping on, because iterators on that container are not going to be informed of your alterations and, as you’ve noticed, that’s quite likely to produce a very different loop and/or an incorrect one. In normal cases, looping on a copy of the container helps, but in your case it’s clear that you don’t want that, as the container will be empty after 50 legs of the loop and if you then try popping again you’ll get an exception.

What’s anything BUT clear is, what behavior are you trying to achieve, if any?! Maybe you can express your desires with a while…?

i = 0
while i < len(some_list):
    print i,                         
    print some_list.pop(0),                  
    print some_list.pop(0)
Answered By: Alex Martelli

This slice syntax makes a copy of the list and does what you want:

l  = range(100)  
for i in l[:]:  
    print i,  
    print l.pop(0),  
    print l.pop(0)
Answered By: thethinman

I guess this is what you want:

l  = range(100)  
index = 0                       
for i in l:                         
    print i,              
    try:
        print l.pop(index+1),                  
        print l.pop(index+1)
    except IndexError:
        pass
    index += 1

It is quite handy to code when the number of item to be popped is a run time decision.
But it runs with very a bad efficiency and the code is hard to maintain.

Answered By: York Chang

Use a while loop that checks for the truthfulness of the array:

while array:
    value = array.pop(0)
    # do some calculation here

And it should do it without any errors or funny behaviour.

Answered By: Eugene Eeo
Categories: questions Tags: , , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.