你想实现一个自定义迭代模式,跟普通的内置函数比如 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 语句会自动处理这些细节,所以你无需担心。