生成器

2018-01-10  本文已影响0人  IT孤独者

这个语法特性需要单独写一篇文章介绍。

在C/C++中是没有所谓的生成器的语法特性,简单的说就是没有 yield 关键字。那么就会有这个疑问,既然Python已经符合基本的语言完备性,为什么还要增加这个生成器的特性,究其原因就是因为生成器可以节省内存。

Python的内存管理是由解释器来完成的,对于列表这个及其特殊的内建数据结构,本身的特性是需要一片连续的地址空间(或者说逻辑上是连续的地址空间),导致如果需要循环一个大的列表,就必须先开辟这样的地址空间,这对于性能和空间来说都是挑战,有了生成器,我们只要能用表达式或者函数表示这个列表的生成规则,那么我们就可以一次只生成一个数据对象,供 for 语句消费,那么内存空间的需求就会减少很多。所以,生成器并不是书写Python 代码必要的工具,但是,这个工具有特殊的意义,就是对于列表空间的优化。我们知道不论是树形结构还是图形结构,我们都会涉及线性化的问题,简单说就是遍历的问题,而遍历的结果往往就是一个列表,所以,不用我说你也应该明白,生成器对于遍历某种数据结构的意义。

下面是一个合并两个有序序列,再对整个序列进行迭代的例子

def merge(a, b):
    i, j = 0, 0
    len_a, len_b = len(a), len(b)
    while True:
        if i >= len_a and j >= len_b:
            break
        if i >= len_a:
            yield b[j]
            j += 1
        elif j >= len_b:
            yield a[i]
            i += 1
        elif a[i] < b[j]:
            yield a[i]
            i += 1
        else:
            yield b[j]
            j += 1

if __name__ == '__main__':
    for item in merge([1, 4, 7, 10], [2, 5, 6, 11]):
        print(item, end=',')
上一篇 下一篇

猜你喜欢

热点阅读