生成器

生成器

生成器，利用迭代器，我们可以在每次迭代获取数据时（通过next()方法）按照特定的规律进行生成。但是我们在实现一个迭代器时，关于当前迭代的状态需要我们自己记录，进而才能根据当前的状态生成下一个数据。为了达到记录当前状态，并配合next()函数进行迭代使用，可以采用更简便的语法。

即 生成器（generator），生成器是一种特殊的迭代器，它比迭代器更优雅。

创建一个生成器

列表[] —> ()

li = [x**2 for x in range(6)]
print(li)

gen = (x**2 for x in range(6))
print(gen)

print("通过next()函数取得下一个值")
print(next(gen))

print("通过for遍历")
gen = (x**2 for x in range(6))
for i in gen:
    print(i,end=" ")
>>>通过next()函数取得下一个值
0
通过for遍历
0 1 4 9 16 25    

生成器函数

在函数中如果出现了yield关键字，那么该函数就不再是一个普通函数而是一个生成器函数。

def foo():
    yield 1  
    yield 2
    return   
    yield 3  #  当程序遇到return，return后的语句不会在执行。因此报错

f = foo()
print(next(f))  #  程序会停留在对应yield后的语句
print(next(f))
print(next(f))  #  当程序遇到return，return后的语句不会在执行。因此报错
>>>1
2

next 和 yield进行匹配。如果遇到return，

return后的语句不会再执行，直接抛出StopIteration，终止迭代。

在一个生成器中，如果return后有返回值，那么这个值就是异常的说明，而不是函数的返回值。

生成器的支持的方法

• close()

• 手动关闭生成器函数，后面调用会直接引起StopIteration异常

• send()

  • #x = yield y 语句的含义：send（）的作用就是使x赋值为其所传送的值(send的参数)，然后让生成器执行到下一个yield。

  • 如果生成器未启动，则必须在使用send（）前启动生成器，而启动的方法可以是gen.next(),也可以使gen.send(None)执行到第一个yield处。之后就可以使用send(para) 不断的传入值了。

  • 如果是已启动，则send(para)的作用就是给x

    赋值为发送的值（send的参数），然后，让生成器执行到下一个yield

    def gen():
        value = 0
        while True:
            receive = yield value
            if receive == "e":
                break
            value = "got:%s"% receive
    # send()的作用就是使receive赋值为其所传送的值，然后让生成器执行到下一个yield
    g = gen()
    print(g.send(None))
    print(g.send("aaa"))
    >>>0
    got:aaa
    

• throw()

  • 手动抛出异常。