How to conditionally skip number of iteration steps in a for loop in python?

Question

We have a list item_list,

item_list = ["a", "b", "XYZ", "c", "d", "e", "f", "g"]

We iterate over its items with a for loop, if item is "XYZ", skip items "c", "d", "e" and continue with "f":

for item in item_list:
    if item == "XYZ":
       do_something()
       skip_3_items() ----> skip items "c", "d", "e"
    else:
        do_something_else()

What could be the most pythonic way to achieve this?

You want to skip those particular items or any next three items in the list ? — jramirez
– jramirez, Commented Jan 16, 2014 at 18:00
You could use a counter and continue until it's 0, but i'm pretty sure the most pythonic way would be rethinking your loop entirely — loopbackbee
– loopbackbee, Commented Jan 16, 2014 at 18:01

kindall · Accepted Answer · 2014-01-16 18:08:46Z

7

list_iter = iter(item_list)

for item in list_iter:
    if item == "XYZ":
        do_something()
        for _ in range(3):   # skip next 3 items
            next(list_iter, None)

# etc.

Basically, rather than iterating over the list directly, you create an abstraction for it called an iterator and iterate over that. You can tell the iterator to advance to the next item by calling next(...) which we do three times to skip the next three items. The next time through the loop, it picks up at the next item after that.

edited Jan 16, 2014 at 18:08

answered Jan 16, 2014 at 18:01

kindall

185k36 gold badges291 silver badges321 bronze badges

Sign up to request clarification or add additional context in comments.

7 Comments

alwbtc Over a year ago

What does next(list_iter) do?

loopbackbee Over a year ago

@alwbtc it advances the iterator one position

Steinar Lima Over a year ago

You should put those next statements in a try ... except StopIteration

loopbackbee Over a year ago

@alwbtc if you're new to iterators, you should read this

DSM Over a year ago

Instead of catching a StopIteration, since you're ignoring the value anyway, you can simply use the default arg, e.g. next(list_iter, None)

|

inspectorG4dget · Accepted Answer · 2014-01-16 18:09:56Z

2

Since nobody has mentioned a while loop, I will:

item_list = ["a", "b", "XYZ", "c", "d", "e", "f", "g"]
i = 0
while i < len(item_list):
    item = item_list[i]
    if item == "XYZ":
        do_something()
        i += 3
    else:
        do_something_else()
    i += 1

answered Jan 16, 2014 at 18:09

inspectorG4dget

115k30 gold badges159 silver badges253 bronze badges

1 Comment

lsoranco Over a year ago

I believe no one has mentioned because it is usually very "non-pythonic" to use while. The whole point of using iterators is to get away from counting integers and stuff. Your code works, of course

damienfrancois · Accepted Answer · 2014-01-16 18:11:04Z

1

Use an iterator:

$ cat t.py 
item_list = ["a", "b", "XYZ", "c", "d", "e", "f", "g"]

it = iter(item_list)
for item in it:
    if item == "XYZ":
       print item
       for _ in range(3):
           next(it, None)
    else:
        print item

This gives:

$ python t.py 
a
b
XYZ
f
g

edited Jan 16, 2014 at 18:11

answered Jan 16, 2014 at 18:04

damienfrancois

60.4k9 gold badges116 silver badges128 bronze badges

Comments

Abhijit · Accepted Answer · 2014-01-16 18:12:43Z

1

I would split the processing it two parts for sake of readability

>>> def foo(item_list,key = "XYZ", skip = 3):
    from itertools import takewhile, islice
    def do_something():
        return "do_something()"
    def do_something_else():
        return "do_something_else()"
    it = iter(item_list)
    for items in takewhile(lambda e: e != key, it):
        print items, do_something_else()
    print do_something()
    it = islice(it,skip, None)
    for items in it:
        print items, do_something_else()


>>> foo(item_list)
a do_something_else()
b do_something_else()
do_something()
f do_something_else()
g do_something_else()

answered Jan 16, 2014 at 18:12

Abhijit

64k20 gold badges143 silver badges209 bronze badges

Comments

Chris Arena · Accepted Answer · 2014-01-16 18:13:12Z

0

I'd store a counter that handles skipping items.

def skipper(item_list):
    skip_count = 0
    for item in item_list:
        if item == "XYZ":
            skip_count = 3
        else:
            if skip_count:
                skip_count -= 1
            else:
                # do_something()
                print item,

Example:

In [23]: item_list
Out[23]: ['a', 'b', 'XYZ', 'c', 'd', 'e', 'f', 'g']

In [24]: skipper(item_list)
a b f g

edited Jan 16, 2014 at 18:13

answered Jan 16, 2014 at 18:03

Chris Arena

1,65015 silver badges18 bronze badges

Comments

roippi · Accepted Answer · 2014-01-16 18:17:34Z

0

I'd probably write it out like this, personally:

xyz_i = item_list.index('XYZ')

do_something('XYZ') #or do_something(item_list[xyz_i]) but.. just save yourself the list lookup

for x in item_list[xyz_i+4:]:
    do_something_else(x)

edited Jan 16, 2014 at 18:17

answered Jan 16, 2014 at 18:05

roippi

26k4 gold badges52 silver badges75 bronze badges

Collectives™ on Stack Overflow

How to conditionally skip number of iteration steps in a for loop in python?

6 Answers 6

7 Comments

1 Comment

Comments

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

6 Answers 6

7 Comments

1 Comment

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related