Python列表推导式与生成器

愿你一直充满无限的潜力


列表推导式

>>> [x+1 for x in range(10) ]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

从上面的代码可以看出,列表推导式是生成一个列表。

生成器

>>> g=(x+1 for x in range(10))
>>> g
 at 0x7f3af91b7570>
>>> next(g)
1
>>> next(g)
2
>>> next(g)
3
>>> next(g)
4
>>> next(g)
5
>>> next(g)
6
>>> next(g)
7
>>> next(g)
8
>>> next(g)
9
>>> next(g)
10
>>> next(g)
Traceback (most recent call last):
  File "", line 1, in 
StopIteration
>>>

从上面的代码可以看出,生成器的形式和列表推导式差不多,只是括号的不同。

本质区别是列表推导式是一次性生成,而生成器是通过相应的算法推导出来的值,在数据量较大的时候,你不可能一次性生成那么大的列表,生成器是最好的选择。

例如:用生成器产生斐波拉契数列

def fibs(max):
    n,a,b=0,0,1
    while n < max:
        yield b
        a,b = b,a+b
        n+=1

for n in fibs(6):
    print(n)

你可能感兴趣的:(Python列表推导式与生成器)