生成器

1、生成器与yield

在函数内一旦存在yield关键字,调用函数并不会执行函数体代码
会返回一个生成器对象,生成器即自定义的迭代器

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

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

运行  生成器.__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)

2、yield表达式

函数内有yield,那么运行函数就可以拿到一个生成器对象,同时yield用表达式形式为变量传值,如:x=yield 返回值

一:yield返回的生成器对象可以通过方法send()为yield赋值,并传给变量 x
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(['一根骨头','aaa'])
# g.send('肉包子')
# g.send('一同泔水')
# g.close()
# g.send('1111') # 关闭之后无法传值


# 二:定义一个列表,可以将send传的值保存到列表中,并返回
def dog(name):
    food_list=[]
    print('道哥%s准备吃东西啦...' %name)
    while True:
        # x拿到的是yield接收到的值
        x = yield food_list # x = '肉包子'
        print('道哥%s吃了 %s' %(name,x))
        food_list.append(x) # ['一根骨头','肉包子']

g=dog('alex')
res=g.send(None)  # next(g)
print(res)

res=g.send('一根骨头')
print(res)

res=g.send('肉包子')
print(res)
# g.send('一同泔水')



#返回值与send()
def func():
    print('start.....')
    x=yield 1111  # x='xxxxx'
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    print('哈哈哈啊哈')
    yield 22222

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

res=g.send('xxxxx')
print(res)

总结yield:

有了yield关键字,我们就有了一种自定义迭代器的实现方式。yield可以用于返回值,但不同于return,函数一旦遇到return就结束了,而yield可以保存函数的运行状态挂起函数,用来返回多次值

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值