Python yield生成器

Python 中可以通过列表生成式，创建列表，并且可以通过列表迭代访问列表里的每一个元素，如果元素数量很大，而且只需访问前几个元素，使用列表将会造成很大的浪费，生成器就是为了解决这种问题而设计的。

创建generator的方法

1. 把[]改成()

alist = [x*x for x in range(5)]
print(alist)
blist = (x*x for x in range(5))
print(blist)
[0, 1, 4, 9, 16]
<generator object <genexpr> at 0x1056b6280>

2. yield关键字

def func():
    a = 1
    while a < 6:
        b = range(a)
        yield b
        print ("a is {}".format(a))
        a = a + 1

3. 使用方式

yield生成器每次返回一个元素，下一个迭代继续执行当前语句的下一条语句，直到碰到return或者yield语句

def func():
    a = 1
    while a < 6:
        b = range(a)
        yield b
        print ("a is {}".format(a))
        a = a + 1


o = func()
print(o.next())
[0]  # 输出[0]函数到yield语句即返回第一次生成的结果，下次从下一条语句继续执行
print(o.next())
a is 1
[0, 1]     # 此时a=1,列表的值为[0, 1]
print (o.next())
o = func()
for i in o:             # 也可以使用for 循环访问
    print(i)
[0]
a is 1
[0, 1]
a is 2
[0, 1, 2]
a is 3
[0, 1, 2, 3]
a is 4
[0, 1, 2, 3, 4]
a is 5

优点

生成器不会像列表一样把所有元素都放在内存空间里，而是需要时生成一个元素，这样的做法节省了内存空间。在读取大文件时可以尝试采用这种方法。