4.3使用生成器创建新的迭代模式

问题:

​ 你想实现一个自定义迭代模式,跟普通的内置函数比如range(),reverse的()不一样。

解决方案:

​ 如果你想实现一种新的迭代模式,使用一个生成函数来定义它。下面是 一个生产某个范围内浮点数的生成器。

def frange(start, stop, increment):
    x = start
    while x < stop:
        yield x
        x += increment


for n in  frange(1, 10, 3):
    print(n)
print(list(frange(1, 10, 3)))#->[1, 4, 7]

讨论:

​ 一个函数中需要有一个yeild语句即可将其转换为一个生成器。跟普通函数不同的是,生成器只能用于迭代操作。下面是一个实验,向你展示这样的函数底层工作机制。

def countdown(n):
    print('Staring to count from', n)
    while n > 0:
        yield n
        n-=1
    print('Done!')

c=countdown(3)
print(c)
print(next(c))# ->3
print(next(c))# ->2
print(next(c))# ->1
print(next(c))
'''
line 25, in <module>
StopIteration

'''

​ 一个生成器函数主要特征是它只会回应在迭代中使用到的next操作,一旦生成器函数返回退出,迭代终止,我们在迭代中通常使用的for语句会自动处理这些细节,所以你无需担心。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值