首页 » 排名链接 » 使用生成器创建新的迭代模式(生成器函数迭代是一个你想)

使用生成器创建新的迭代模式(生成器函数迭代是一个你想)

admin 2024-11-22 11:38:52 0

扫一扫用手机浏览

文章目录 [+]

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

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

def custom_generator(start, stop, incremental): while start < stop: yield start start += incremental

为了使用这个函数,你可以用 for 循环迭代它或者使用其他接受一个可迭代对象的函数 (比如 sum() , list() 等)。
示例如下:

使用生成器创建新的迭代模式(生成器函数迭代是一个你想) 排名链接
(图片来自网络侵删)

for item in custom_generator(1, 5, 0.5): print(item)"""11.52.02.53.03.54.04.5"""print(list(custom_generator(1, 5, 0.5)))# "[1, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5]"

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

def custom_down(num): print('Starting to down...') while num > 0: yield num num -= 1 print('Done!!!!')down = custom_down(3)print(down)# "<generator object custom_down at 0x00000289BF0F44F8>"print(next(down))"""Starting to down...3"""print(next(down))# "2"print(next(down))# "1"# print(next(down))# 报错 StopIteration

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

标签:

相关文章