Python中的yield、next和send

当定义的方法和函数中有 yield 关键字,这个执行这个函数或者方法就将返回一个生成器对象,这个生成器对象可以被for循环迭代,也可以手动执行next或者send方法精准控制这个生成器的内部执行,我们看如下代码就可以简单理解 yield,next和send之间的关系和用法:

#!/usr/bin/env python
import types

def gen():
        print 'enter'
        a = yield 1 #StateA 
        print 'next'
        print a or 'a'
        b = yield 2 #StateB 
        print b or 'b'
        print 'next again'

g = gen()
print(type(g))
print('g isinstance types.GeneratorTyp : '+ str(isinstance(g, types.GeneratorType)))
print(g.next())

i=0
while(i>=0):
    i += 1
    print 'i:'+str(i)
    try:
        print(g.send('x'))
    except StopIteration:
        break

print('--------------')

g2 = gen()
print(type(g2))

for x in g2:
    print(x)


执行上述代码将打印如下信息:

<type 'generator'>
g isinstance types.GeneratorTyp : True
enter

i:1
next
x

i:2
x
next again
--------------
<type 'generator'>
enter

next
a

b
next again

详细解释一下执行过程
1、定义生成器函数
def gen()
2、获得迭代器对象,并赋值给g,同时我们打印g的类型,返回generator,可以看到我们类型判断为 generator
g = gen()
3、执行迭代器到到第一个yield处,此处打印1,然后迭代器就挂起了,等待下一次调用,我们暂时命名此出挂起状态为StateA。
print(g.next())
4、进入while循环,第一次执行循环,打印“i:1”,然后执行如下代码
print(g.send('x'))
上述代码将继续 StateA 状态开始执行,首先将字符串“x”发送到迭代器中,然后迭代器开始运行,将字符串“x”复制给变量a,然后打印出:

print 'next'  #next
print a or 'a' #x

接着运行到下一个 yield,返回yield值“2”到send函数,于是在while循环中就打印出了“2”,这时迭代器函数的状态运行到StateB
5、再次进入while循环,send函数将字符串“x”传递进迭代器,赋值给b,这样迭代器就打印出

print b or 'b'      #x
print 'next again'  #next again

然后由于迭代器中没有 yield 操作,将抛出一个 StopIteration 异常,我们接受到 StopIteration 异常就终止循环。
6、通过使用for循环来迭代一个迭代器对象更简单,不需要判断 StopIteration 异常就可以获取yield传出来的值,直接将其打印出

转载:Python中的yield,next和send

    原文作者:mihope
    原文地址: https://www.jianshu.com/p/16d91f501be1
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞