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
<generator object <genexpr> 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 "<stdin>", line 1, in <module>
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)