Python生成器介绍（原理+实践）

本篇博文讲解生成器，主要介绍2个方面，其一为原理，其二为完整代码运行及注解说明。

一.生成器原理

生成器却不同，它可以实现在迭代的同时生成元素。

也就是说，对于可以用某种算法推算得到的多个数据，生成器并不会一次性生成它们，而是什么时候需要，才什么时候生成。

不仅如此，生成器的创建方式也比迭代器简单很多，大体分为以下 2 步：

定义一个以 yield 关键字标识返回值的函数；
调用刚刚创建的函数，即可创建一个生成器；

举个例子：

def intNum():
print("开始执行")
for i in range(5):
yield i
print("继续执行")
num = intNum()

由此，我们就成功创建了一个 num 生成器对象。显然，和普通函数不同，number() 函数的返回值用的是 yield 关键字，而不是 return 关键字，此类函数又成为生成器函数。

和 return 相比，yield 除了可以返回相应的值，还有一个更重要的功能，即每当程序执行完该语句时，程序就会暂停执行。不仅如此，即便调用生成器函数，Python 解释器也不会执行函数中的代码，它只会返回一个生成器（对象）。

要想使生成器函数得以执行，或者想使执行完 yield 语句立即暂停的程序得以继续执行，有以下 2 种方式：

通过生成器（上面程序中的 num）调用 next() 内置函数或者 __next__() 方法；
通过 for 循环遍历生成器。

例如，在上面程序的基础上，添加如下语句：

#调用 next() 内置函数
print(next(num))
#调用 __next__() 方法
print(num.__next__())
#通过for循环遍历生成器
for i in num:
print(i)

程序执行结果为：

开始执行
0
继续执行
1
继续执行
2
继续执行
3
继续执行
4
继续执行

这里有必要给读者分析一个程序的执行流程：
1) 首先，在创建有 num 生成器的前提下，通过其调用 next() 内置函数，会使 Python 解释器开始执行 intNum() 生成器函数中的代码，因此会输出“开始执行”，程序会一直执行到yield i，而此时的 i==0，因此 Python 解释器输出“0”。由于受到 yield 的影响，程序会在此处暂停。

2) 然后，我们使用 num 生成器调用 __next__() 方法，该方法的作用和 next() 函数完全相同（事实上，next() 函数的底层执行的也是 __next__() 方法），它会是程序继续执行，即输出“继续执行”，程序又会执行到yield i，此时 i==1，因此输出“1”，然后程序暂停。

3) 最后，我们使用 for 循环遍历 num 生成器，之所以能这么做，是因为 for 循环底层会不断地调用 next() 函数，使暂停的程序继续执行，因此会输出后续的结果。

注意，在 Python 2.x 版本中不能使用 __next__() 方法，可以使用 next() 内置函数，另外生成器还有 next() 方法（即以 num.next() 的方式调用）。

除此之外，还可以使用 list() 函数和 tuple() 函数，直接将生成器能生成的所有值存储成列表或者元组的形式。例如：

num = intNum()
print(list(num))
num = intNum()
print(tuple(num))

程序执行结果为：

开始执行
继续执行
继续执行
继续执行
继续执行
继续执行
[0, 1, 2, 3, 4]
开始执行
继续执行
继续执行
继续执行
继续执行
继续执行
(0, 1, 2, 3, 4)

通过输出结果可以判断出，list() 和 tuple() 底层实现和 for 循环的遍历过程是类似的。

相比迭代器，生成器最明显的优势就是节省内存空间，即它不会一次性生成所有的数据，而是什么时候需要，什么时候生成。

二.完整代码说明：

# 带有参数的生成器，参数为k
def intNum(k):
    print("开始执行")
    n=0
    while n<4:
        n=n+1
        yield n+5+k
        print("继续执行")
num = intNum(3)
# 说明参数k=3保持不变，每次迭代都不会改变，
# 而n将会在每次迭代中一直+1，当n<10满足条件
# 就终止迭代。
for i in num:
    # 这里i是num函数返回的值，即n+5+k
    print(i)
# 注：第一运行将执行wile上面的代码，随后在yield处停止，
#     当再次迭代时将执行yield后面的代码（print("继续执行")），
#     然后在从while处执行直到遇到yield后，又重复上述步骤执行

运行结果：

原理主要借鉴（很详细）：http://c.biancheng.net/view/2393.html

posted @ 2020-02-13 22:29 tangjunjun 阅读(1045) 评论(0) 编辑收藏举报

刷新页面返回顶部

tangjunjun

Python生成器介绍（原理+实践）

公告