[00404]使用生成器创建新的迭代模式

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

一、 解决问题

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

二、解决方案

生成器函数

三、代码说明

#!/usr/bin/env python

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

for n in frange(10, 1.0, 0.5):
    print(n)

"""
一个函数如果存在yield语句，那么则其就是一个生成器。
生成器只能用于迭代操作。（不同于其他普通的函数）

"""

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

c = countdown(3)
print (c)

print(next(c))
print(next(c))
print(next(c))
print(next(c)) #打印Done!并出错

四、关联知识

五、总结

1. 一个生成器函数主要特征，它会回应在迭代器中使用next操作。
2. 一旦生成器函数返回退出，迭代终止。

六、代码地址

github地址:https://github.com/weichen666/python_cookbook
目录/文件：fourth_selection/learn_iter_gender.py

七、参考