Python复习之生成器 generator

生成器 generator

def h(): 
        print "hello" 
        m = yield 5 
        print "m:", m 
        d = yield 12 
        print "d:", d 
        print "world"

结果

c = h()
c.next()  # hello   # 5
c.next()  # m: None   # 12
c.next()  # d: None   # world   # StopIteration

yield类似于函数里面的return 返回yield后面的值 如:yield 5 会返回5
那么 m = yield 5 这个m的值到底是多少? 现在从上面的结果看出来是 None。

那么这个m值怎么来?

send()方法和next()方法的功能有很大的重合:都是让函数体继续向下执行直到遇到下一个yield语句并挂起。而不同点是send(msg)方法会给生成器发送变量msg,并作为yield表达式的返回值赋值给其前面的变量,也就是h()中的m和d.
next()就等价于send(None)。

接着往下走

c = h()
c.send("I am m") 
# TypeError: can't send non-None value to a just-started generator

可以看到出错了,正如错误提示所说,对于刚开始的生成器是不能send 一个non-None值的。由此也可以知道,必须要先yield,才能send(msg)。再来:

c = h()
c.send(None) # hello   # 5
c.send("I am m") # m: I am m   # 12
c.send("I am d")  # d: I am d   # world   # StopIteration

由上可以总结出,yield 后面的内容和前面的赋值其实没什么关系,前面的赋值是在yield 完成以后由send()决定的。
next启动生成器,遇到yield则停止

posted @ 2016-12-16 21:53  Dus  阅读(156)  评论(0编辑  收藏  举报