Python中的迭代、生成和yield关键字

发表于 更新于

0X00 可迭代对象

Python中的列表，元组，字典，文件都是可迭代对。可迭代对象简单地说就是可以用for i in xxx:来遍历的对象。

1
2
3
4
5
6
7
8
9
10
my_list = [1, 2, 3, 4, 5, 6, 7]
for i in my_llist:
    print i

my_dict = {
    'a': u'苟利国家生死以',
    'b': u'岂因祸福避趋之'
}
for i in my_dict:
    print i

不过如果数据量非常非常庞大的时候，会很影响程序的性能。这种时候就可以使用生成器来解决这个问题。

0X01 生成器

生成器的用法和普通的可迭代对象差不多，最大的特点就是：“用的时候才去计算”。这里写一个简单的例子，演示一下情况。第一种生成超大列表的方式要逐项计算完才算弄出来了这个10000长的列表，而后者是生成器，只是声明了怎么算，并没有真的去算，所以在速度上才是完全不能比的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/usr/bin/env python
# coding=utf-8

import time

if __name__ == '__main__':
    start = time.time()
    [x**x for x in range(10000)]    # 这里生成的是一个超大的列表
    end = time.time()
    print end - start

    start = time.time()
    (x**x for x in range(10000))    # 这里生成一个超大的生成器（注意看，括号不一样）
    end = time.time()
    print end - start

运行结果如下：

1
2
3
shawn in ~ λ python hello.py
6.76475715637
0.000135898590088

0X02 yield关键字

那是时候自己生成弄一个生成器出来了，Python中提供的yield关键字就是用来干这个的，通过这个关键字可以创造自己的生成器。
还是上面同样的例子，稍加改动

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/usr/bin/env python
# coding=utf-8

import time

def create_list():
    # 创建一个普通的列表
    my_list = []
    for x in range(10000):
        my_list.append(x**x)
    return my_list

def test_yield():
    # 使用yield
    for x in range(10000):
        yield x**x

if __name__ == '__main__':
    start = time.time()
    a = create_list()
    end = time.time()
    print end - start

    start = time.time()
    a = test_yield()
    end = time.time()
    print end - start

运行起来看到的时间差距和刚刚演示的也差不多

1
2
3
shawn in ~ λ python hello.py
6.45007514954
0.0080668926239