首页 > 其他 > 详细

生成器

时间:2020-05-29 00:09:06      阅读:51      评论:0      收藏:0      [点我收藏+]

什么是生成器

生成器就是自定义的迭代器

如何得到自定义的迭代器

在函数内一旦存在yield关键字,调用函数并不会执行函数体代码

会返回一个生成器对象,生成器即自定义的迭代器

def func():
    print(‘第一次‘)
    yield 1
    print(‘第二次‘)
    yield 2
    print(‘第三次‘)
    yield 3
    print(‘第四次‘)


g=func()
print(g)
# 生成器就是迭代器
g.__iter__()
g.__next__()

会触发函数体代码的运行,然后遇到yield停下来,将yield后的值当做本次调用的结果返回

res1=g.__next__()
print(res1)


res2=g.__next__()
print(res2)

res3=g.__next__()
print(res3)

res4=g.__next__()


len(‘aaa‘) # ‘aaa‘.__len__()

next(g)    # g.__next__()
iter(可迭代对象)     # 可迭代对象.__iter__()

应用案列

def my_range(start,stop,step=1):
    # print(‘start...‘)
    while start < stop:
        yield start
        start+=step
    # print(‘end....‘)


g=my_range(1,5,2) # 1 3
print(next(g))
print(next(g))
print(next(g))

for n in my_range(1,7,2):
    print(n)

总结yield

有了yield关键字,我们就有了一种自定义迭代器的实现方式。

yield可以用于返回值,但不同于return,函数一旦遇到return就结束了,而yield可以保存函数的运行状态挂起函数,用来返回多次值

生成器的高级玩法之yield挂起函数:yield的表达式形式

def dog(name):
    print(‘道哥%s准备吃东西了...‘ % name)
    while True:
        # x拿到的是yield接收到的值
        x = yield  # x = ‘一根骨头‘
        print(‘道哥%s吃了 %s‘ % (name, x))


g = dog(‘alex‘)

g.send(None)  # 等同于next(g),先完成初始化

g.send(‘一根骨头‘)
# g.send(‘一根骨头‘,‘肉包子‘)    # 不能传2个值,会报错:TypeError: send() takes exactly one argument (2 given)
g.send([‘一根骨头‘,‘肉包子‘])    # 可以放到列表里传输,列表算是一个值
g.send(‘肉包子‘)
g.send(‘一桶泔水‘)
# g.close()
# g.send(‘1111‘) # 关闭之后无法传值

综合应用

def dog(name):
    food_list = []
    print(‘大哥%s准备吃东西了...‘ % name)
    while True:
        # x拿到的是yield接收到的值
        # x = yield 111 # x = [‘一根骨头‘,‘肉包子‘]
        x = yield food_list # x = [‘一根骨头‘,‘肉包子‘]
        print(‘大哥%s抽了 %s‘ % (name, x))
        food_list.append(x)


g = dog(‘egon‘)

res = g.send(None)  # 等同于next(g),先完成初始化
print(res)

res = g.send([‘一个鞭炮‘,‘一根烟‘])
print(res)

res = g.send(‘一个地雷‘)
print(res)
def func():
    print(‘start...‘)
    x = yield 111
    print(‘H哈哈‘)
    print(‘H哈哈‘)
    print(‘H哈哈‘)
    yield 222

g = func()
res = next(g)
print(res)

res = g.send(‘xxxx‘)
print(res)

生成器

原文:https://www.cnblogs.com/zhww/p/12983961.html

踩
(0)
赞
(0)
   
举报
评论 一句话评论(0)
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!