CODEWITHSUNDEEP
pythongeneratoryieldyield-keyword
neuront
neuront

Reputation: 9622

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

Could anyone please explain that? Thanks in advance.

Upvotes: 11

Views: 5339

Answers (1)

Blender
Blender

Reputation: 298402

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'

Upvotes: 11

Related Questions