生成器
生成器概念
生成器对象其实就是自定义的迭代器,其本质上是迭代器,也就是说生成器就是一个可以进行迭代生成的工具
生成器对象
函数体代码内含有yield关键字的时候,第一次调用函数的时候不会执行函数体代码,而是会返回一个结果,该结果就是生成器。也就是第一次调用函数会将函数变成生成器。生成器对象内部含有__iter__
和__next__
方法
def index():
yield 123
yield 666
res = index() # 调用不执行函数体代码,将函数变成生成器对象
print(res) # <generator object index at 0x00000204B99BFEB8>
print(res.__next__()) # 123 # 然后对执行后的结果调用__next__方法执行代码
print(res.__next__()) # 666
print(res.__next__()) # 报错 执行完再执行会报错
yield关键字
- yield 关键字的功能
- 函数体代码内含有yield时,会将函数变成生成器
- 多个yield关键字存在函数体代码内时,调用__next__执行到yield关键字的时候会将yield关键字后面的数据返回出去,并将代码停留在这一步。再次执行__next__时是从暂停的这一步继续进行执行的。执行到代码结束,也会报错。
- 接受外界的传值
def h(name):
while True:
game = yield
print(f'{name} is paling {game}')
res = h('oliver')
res.__next__() # 先初始化一次,等待给yield传值
res.send('piano') # 接受外界传值给yield 结果:oliver is paling piano
res.send('pingpang') # 接受外界传值给yield 结果:oliver is paling pingpang
生成器表达式
res = (i for i in 'oliver')
print(res)
print(res.__next__()) # o
print(res.__next__()) # l
print(res.__next__()) # i
print(res.__next__()) # v
print(res.__next__()) # e
print(res.__next__()) # r
print(res.__next__()) # 报错