python 生成器

生成器

官方定义:

• 在python中,使用了yield的函数被称为生成器(generator).
• 跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器.
• 在调用程序的过程中,每次遇到yield时函数会暂停并保存当前所有的运行信息,返回yield的值,并在下一次执行next()方法时从当前位置继续运行
• 调用一个生成器函数,返回的是一个迭代器对象

---

我是这么理解的 , 但凡函数内包含了yield关键字,调用函数就不会执行函数代码,而是得到一个返回值.改返回值就是生成器对象

创建生成器(第一种)

l = [x * 2 for x in range(5)]

print('L',l)

g = (x * 2 for x in range(5))

print("g",g)

结果:
区别就是一个是[],一个是(),L打印结果,g打印生成器对象
而对于生成器g，我们可以按照迭代器的使用方法来使用，即可以通过next()函数、for循环、list()等方法使用。

---

创建生成器(第二种)

def gen():
    i = 0
    while i < 5:
        yield i
        i += 1


g = gen()
print(g)

结果:

总结 yield next调用执行顺序

• 使用了yield关键字的函数不再是函数，而是生成器。
• 使用next调用生成器，执行顺序：
• 如果是第一次执行，则从def代码块的开始部分执行，直到遇到yield为止，并且把yield关键字后的数值返回，当做next()的返回值, yield在这里有了return返回值的作用。
• 如果不是第一次执行，则从上一次暂停的位置执行(即从上一次yield关键字的下一个语句开始执行)，直到遇到下一次yield为止，，并且把yield关键字后的数值返回，当做next()的返回值

生成器里面有return

额…return的值必须是报错的时候才能拿到

def gen():
    i = 0
    while i < 5:
        yield i
        i += 1

    return "获取完毕"

g = gen()
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))

可以自己运行看一下结果就行 (使用for循环是拿不到return的值的)

因为报错的时候才能拿到return , 所有捕获一下错误

while True:
    try:
        print(next(g))
        print(next(g))
        print(next(g))
        print(next(g))
        print(next(g))
        print(next(g))


    except StopIteration as e:
        print("生成器返回值:%s" % e.value)
    break

就可以正常拿到了

---

send唤醒

def gen():
    i = 0
    while i < 5:
        temp = yield i
        print(i)
        print('temp',temp)
        i+=1


f = gen()
next(f)
next(f)
next(f)
f.send("我是send发来的数据")
next(f)

使用send()函数的一个好处是可以在唤醒的同时向断点处传入一个附加数据。

send发送的数据是给yield关键字前面变量发,要是打印next(f),得到的是yield返回值 , 触发条件不一样