生成器

生成器

生成器，利用迭代器，我们可以在每次迭代获取数据时(通过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(next(gen))
print('通过for 遍历')    #前面迭代取了2个
for i in gen:
    print(i,end=' ')
    
[0, 1, 4, 9, 16, 25]
<generator object <genexpr> at 0x0000021D35DDEF68>
通过next()函数取得下一个值
0
1
通过for 遍历
4 9 16 25 

生成器函数

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

def foo():
    yield 1
    yield 2
print(next(f))   #1
print(next(f))   #2

#若 加入return
def foo():
    yield 1
    return 'Error'
    yield 2
#1
#StopIteration: Error
#则  1可以迭代输出  进而return返回   2无法进行迭代输出StopIteration

next与yield进行匹配。如果遇到return，之后的语句就不会再执行，直接抛出异常StopIteration，终止迭代。

如果return后有返回值，那么这个值就是异常说明 。

测试：奇数生成器

class Odd:
    def __init__(self):
        self.num = -1
    def __iter__(self):
        return self
    def __next__(self):
        self.num += 2
        return self.num
odd = Odd()
for i in range(6):
    print(next(odd), end=' ')

1 3 5 7 9 11 

生成器支持的方法

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

def gen():
    yield 1
    yield 2
    yield 3
    yield 4
g = gen()
print(next(g))
print(next(g))
g.close()#关闭
print(next(g))
#1 2 报错 回溯错误

• send
  • x = yeild y send()的作用就是使 x赋值，然后生成器执行下一个yield
  • 生成器未启动时 启动方法未 gen.next() 或者 gen.send(None)执行到第一个yield处开始生成
  • 启动后传入 gen.send(para) 则yield para 生成

def gen():
    value = 0
    while True:
        receive = yield value
        if receive == 'e':
            break
        value = 'got: %s' %receive
#send()的作用就是使 receive赋值，然后生成器执行下一个yield
# 生成器未启动时  启动方法未 gen.next() 或者 gen.send(None)执行到第一个yield处开始生成
g = gen()
print(g.send(None))
print(g.send('aaa'))
print(g.send(123))
print(g.send('e'))
# 0
# got:aaa
# got: 123
报错 StopIteration

解析：

1. g.send(None)启动生成器， 程序运行到 yield value终止 返回为 0
2. g.send(‘aaa’) ,程序 从上一步yield value后 If开始运行 send赋值给receive = ‘aaa’ 判断条件后 value=‘got: aaa’ 进而 yield value 返回 got: aaa并终止
3. 同理 g.send(123) 赋值receive = 123, 进入 value = ‘ got: 123’, yield value 终止并返回 got:123
4. g.send('e) if判断break 此时迭代器没有更多的值，则报错。