Python复习之生成器 generator

时间：2023-03-09 16:55:26
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则停止

相关文章