(Python) List index out of range - iteration

user3142764 picture user3142764 · Dec 28, 2013 · Viewed 62.2k times · Source
for i in range(len(lst)):    
   if lst[i][0]==1 or lst[i][1]==1:
        lst.remove(lst[i])
return lst

This gives "IndexError: list index out of range" Why is this happening?

Answer

Tim Pietzcker picture Tim Pietzcker · Dec 28, 2013

You're modifying the list you're iterating over. If you do that, the size of the list shrinks, so eventually lst[i] will point beyond the list's boundaries.

>>> lst = [1,2,3]
>>> lst[2]
3
>>> lst.remove(1)
>>> lst[1]
3
>>> lst[2]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range

It's safer to construct a new list:

return [item for item in lst if item[0]!=1 and item[1]!=1]