Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Python - Understanding the send function of a generator

I'm studying in Python yield and find that yield is not only the way in which generators output a return value but also a way to put values into a generator. For example the following code

def f():
    print (yield),
    print 0,
    print (yield),
    print 1

g = f()
g.send(None)
g.send('x')
g.send('y')

In the global scope it sends value 'x', 'y' to the generator and thus in f it will output x 0 y 1. But I cannot understand

  • There are 2 yields but 3 sends. Why should it send None at the first time?
  • It throws a StopIteration at the last send. Is there any way to avoid this exception?

Could anyone please explain that? Thanks in advance.

like image 459
neuront Avatar asked Dec 05 '12 04:12

neuront


1 Answers

From the documentation:

When send() is called to start the generator, it must be called with None as the argument, because there is no yield expression that could receive the value.

As for the exception, you can't really avoid it. The generator throws this exception when it's done iterating, so instead of avoiding it, just catch it:

g = f()

try:
    g.send(None)
    g.send('x')
    g.send('y')
except StopIteration:
    print 'Done'
like image 82
Blender Avatar answered Sep 29 '22 14:09

Blender