python--生成器协程运算

生成器

我们定义一个如下的生成器：

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

p = put_on("bigberg")

#输出
G:\python\install\python.exe G:/python/untitled/study4/test/double.py

Process finished with exit code 0

　　当我们把一个函数通过yield转换成生成器，直接运行函数是不会出现结果返回的。因为此时函数已经是个生成器了，我们要通过next()来取得值，并且在遇到yield时再次跳出函数。

print(type(p))

#输出
<class ‘generator‘>

　　我们添加next()方法：

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield       #遇到yield中断
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

p = put_on("bigberg")
p.__next__()

#输出
Hi bigberg, 货物来了,准备搬到仓库！

　　此时函数中断在 goods = yield 的地方，当我们再次调用next()函数时，函数只会运行中断以后的内容，即上例中的红色字体部分。

　　我们再添加一个next():

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

p = put_on("bigberg")
p.__next__()
p.__next__()

#输出
Hi bigberg, 货物来了,准备搬到仓库！
货物[None]已经被bigberg搬进仓库了。

　　我们可以看到yield并没有给goods传值，所有货物是 None。

小结：

• 通过yield将函数转换为函数，需要使用next()方法才能运行
• yield只是保留函数的中断状态，再次调用next()会执行yield后面的部分
• yield如果没有返回值，会返回一个None空值

二、send()传值

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

p = put_on("bigberg")
p.__next__()
p.send("瓜子")

#输出
Hi bigberg, 货物来了,准备搬到仓库！
货物[瓜子]已经被bigberg搬进仓库了。

小结：

• __next__()只是调用这个yield，也可以说成是唤醒yield，但是不不会给yield传值。
• send()方法调用yield是，能给yield传值
• 使用send()函数之前必须使用__next__()，因为先要中断，当第二次调用时，才可传值。

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

p = put_on("bigberg")
p.__next__()
p.send("瓜子")
p.send("花生")
p.send("饼干")
p.send("牛奶")

#多次调用send()
Hi bigberg, 货物来了,准备搬到仓库！
货物[瓜子]已经被bigberg搬进仓库了。
货物[花生]已经被bigberg搬进仓库了。
货物[饼干]已经被bigberg搬进仓库了。
货物[牛奶]已经被bigberg搬进仓库了。

三、单线程实现并行效果（协程）

import time

def put_on(name):
    print("Hi {}, 货物来了,准备搬到仓库！".format(name))
    while True:
        goods = yield
        print("货物[%s]已经被%s搬进仓库了。"%(goods,name))

def transfer(name):
    p = put_on(‘A‘)
    p2 = put_on(‘B‘)
    p.__next__()
    p2.__next__()
    print("%s将货物送来了!"%name)
    for i in range(5):
        time.sleep(1)
        print("%s递过来两件货物"%name)
        p.send("瓜子")
        p2.send("花生")

transfer("bigberg")

#输出
Hi A, 货物来了,准备搬到仓库！
Hi B, 货物来了,准备搬到仓库！
bigberg将货物送来了!
bigberg递过来两件货物
货物[瓜子]已经被A搬进仓库了。
货物[花生]已经被B搬进仓库了。
bigberg递过来两件货物
货物[瓜子]已经被A搬进仓库了。
货物[花生]已经被B搬进仓库了。
bigberg递过来两件货物
货物[瓜子]已经被A搬进仓库了。
货物[花生]已经被B搬进仓库了。
bigberg递过来两件货物
货物[瓜子]已经被A搬进仓库了。
货物[花生]已经被B搬进仓库了。
bigberg递过来两件货物
货物[瓜子]已经被A搬进仓库了。
货物[花生]已经被B搬进仓库了。