How to detect generator end explicitly in Python?

Question

I wrote a generator, which returns tuple of values:

import numpy as np

def mygenerator():
    data = np.arange(10)
    data = np.reshape(data, (-1, 2))
    for row in data:
        yield row[0], row[1]

generator = mygenerator()
while True:
    a, b = generator.__next__()
    print("a: ", a, "b: ", b)

and want to use it without for ... in. How detect end of generation then?

Just keep calling `next` on it and manually catch `StopIteration`? — Jon Clements, Sep 12 '17 at 09:44
I know it may be an annoying question, but why can you not use a for loop? Mind showing us a bit more code to see if it is really impossible? — Reut Sharabani, Sep 12 '17 at 09:47
Am also a bit surprised to learn what is probably the [answer](https://stackoverflow.com/a/1966609/4288043) to this question. — cardamom, Sep 12 '17 at 09:49

score 18 · Accepted Answer · answered Sep 12 '17 at 09:45

18

The generator will throw an exception StopIteration when it is exhausted. You can catch the exception with try…except statement and exit loop when it occurs:

while True:
    try:
        a, b = next(generator)
    except StopIteration:
        break
    print("a: ", a, "b: ", b)

BTW, why not for loop? It seems that:

for a, b in generator:
    print("a: ", a, "b: ", b)

does exactly what you want, without cumbersome constructions.

answered Sep 12 '17 at 09:45

Błotosmętek

12,365
18
29

6

This is Stackoverflow: if you write complex example, you told to make it short, if you write short example you told to use simple code suitable for that short example :) – Dims Sep 12 '17 at 10:02
2

I can't use `for ... in` because in my real code I use complex processing of returned values and switching different loops... – Dims Sep 12 '17 at 10:03

score 0 · Answer 2 · answered Apr 30 '22 at 22:56

Another possibility would be by using the default parameter of next built-in function as a breaking condition. Once the generator is exhausted, instead of throwing a StopIteration exception, the default value will be emitted.

# ...

gen = mygenerator()

stop_value = None # or another value
while True:
    v = next(gen, stop_value)
    if v is stop_value:
        break
    print("a: {}, b: {}".format(*v))

score -1 · Answer 3 · edited Jan 23 '21 at 14:03

-1

Stumped into this and suggest do following, generator will exhaust until next() returns None:

import numpy as np

def mygenerator():
    data = np.arange(10)
    data = np.reshape(data, (-1, 2))
    for row in data:
        yield row[0], row[1]

generator = mygenerator()
a, b = next(generator, None)
while a and b:
    print("a: ", a, "b: ", b)
    a, b = next(generator, None)

edited Jan 23 '21 at 14:03

Greenonline

1,265
6
21
27

answered Jan 23 '21 at 03:53

jaydrill

44
7

How to detect generator end explicitly in Python?

3 Answers3