通過(guò)列表生成式，我們可以直接創(chuàng)建一個(gè)列表，但是，受到內(nèi)存限制，列表容量肯定是有限的，而且創(chuàng)建一個(gè)包含100萬(wàn)個(gè)元素的列表，不僅占用很大的存儲(chǔ)空間，如果我們僅僅需要訪問(wèn)前面幾個(gè)元素，那后面絕大多數(shù)元素占用的空間都白白浪費(fèi)了。

所以，如果列表元素可以按照某種算法推算出來(lái)，那我們是否可以在循環(huán)的過(guò)程中不斷推算出后續(xù)的元素呢?這樣就不必創(chuàng)建完整的list，從而節(jié)省大量的空間，在Python中，這種一邊循環(huán)一邊計(jì)算的機(jī)制，稱為生成器：generator

生成器是一個(gè)特殊的程序，可以被用作控制循環(huán)的迭代行為，python中生成器是迭代器的一種，使用yield返回值函數(shù)，每次調(diào)用yield會(huì)暫停，而可以使用next()函數(shù)和send()函數(shù)恢復(fù)生成器。

生成器類似于返回值為數(shù)組的一個(gè)函數(shù)，這個(gè)函數(shù)可以接受參數(shù)，可以被調(diào)用，但是，不同于一般的函數(shù)會(huì)一次性返回包括了所有數(shù)值的數(shù)組，生成器一次只能產(chǎn)生一個(gè)值，這樣消耗的內(nèi)存數(shù)量將大大減小，而且允許調(diào)用函數(shù)可以很快的處理前幾個(gè)返回值，因此生成器看起來(lái)像是一個(gè)函數(shù)，但是表現(xiàn)得卻像是迭代器

python中的生成器

要?jiǎng)?chuàng)建一個(gè)generator，有很多種方法，第一種方法很簡(jiǎn)單，只有把一個(gè)列表生成式的[]中括號(hào)改為()小括號(hào)，就創(chuàng)建一個(gè)generator

舉例如下：

#列表生成式

lis=[x*xforxinrange(10)]

print(lis)

#生成器

generator_ex=(x*xforxinrange(10))

print(generator_ex)

結(jié)果：

[0,1,4,9,16,25,36,49,64,81]

at0x000002A4CBF9EBA0>

那么創(chuàng)建list和generator_ex，的區(qū)別是什么呢?從表面看就是[]和(),但是結(jié)果卻不一樣，一個(gè)打印出來(lái)是列表(因?yàn)槭橇斜砩墒?，而第二個(gè)打印出來(lái)卻是at0x000002A4CBF9EBA0>，那么如何打印出來(lái)generator_ex的每一個(gè)元素呢?

如果要一個(gè)個(gè)打印出來(lái)，可以通過(guò)next()函數(shù)獲得generator的下一個(gè)返回值：

#生成器

generator_ex=(x*xforxinrange(10))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

print(next(generator_ex))

結(jié)果：

Traceback(mostrecentcalllast):

File"列表生成式.py",line42,in

print(next(generator_ex))

StopIteration

大家可以看到，generator保存的是算法，每次調(diào)用next(generaotr_ex)就計(jì)算出他的下一個(gè)元素的值，直到計(jì)算出最后一個(gè)元素，沒(méi)有更多的元素時(shí)，拋出StopIteration的錯(cuò)誤，而且上面這樣不斷調(diào)用是一個(gè)不好的習(xí)慣，正確的方法是使用for循環(huán)，因?yàn)間enerator也是可迭代對(duì)象：

#生成器

generator_ex=(x*xforxinrange(10))

foriingenerator_ex:

print(i)

結(jié)果：

所以我們創(chuàng)建一個(gè)generator后，基本上永遠(yuǎn)不會(huì)調(diào)用next()，而是通過(guò)for循環(huán)來(lái)迭代，并且不需要關(guān)心StopIteration的錯(cuò)誤，generator非常強(qiáng)大，如果推算的算法比較復(fù)雜，用類似列表生成式的for循環(huán)無(wú)法實(shí)現(xiàn)的時(shí)候，還可以用函數(shù)來(lái)實(shí)現(xiàn)。

比如著名的斐波那契數(shù)列，除第一個(gè)和第二個(gè)數(shù)外，任何一個(gè)數(shù)都可以由前兩個(gè)相加得到：

1，1，2，3，5，8，12，21，34.....

斐波那契數(shù)列用列表生成式寫不出來(lái)，但是，用函數(shù)把它打印出來(lái)卻很容易：

#fibonacci數(shù)列

deffib(max):

n,a,b=0,0,1

whilen

a,b=b,a+b

n=n+1

print(a)

return'done'

a=fib(10)

print(fib(10))

a,b=b,a+b其實(shí)相當(dāng)于t=a+b,a=b,b=t，所以不必寫顯示寫出臨時(shí)變量t，就可以輸出斐波那契數(shù)列的前N個(gè)數(shù)字。上面輸出的結(jié)果如下：

仔細(xì)觀察，可以看出，fib函數(shù)實(shí)際上是定義了斐波拉契數(shù)列的推算規(guī)則，可以從第一個(gè)元素開(kāi)始，推算出后續(xù)任意的元素，這種邏輯其實(shí)非常類似generator。

也就是說(shuō)上面的函數(shù)也可以用generator來(lái)實(shí)現(xiàn)，上面我們發(fā)現(xiàn)，print(b)每次函數(shù)運(yùn)行都要打印，占內(nèi)存，所以為了不占內(nèi)存，我們也可以使用生成器，這里叫yield。如下：

deffib(max):

n,a,b=0,0,1

whilen

yieldb

a,b=b,a+b

n=n+1

return'done'

a=fib(10)

print(fib(10))

但是返回的不再是一個(gè)值，而是一個(gè)生成器，和上面的例子一樣，大家可以看一下結(jié)果：

那么這樣就不占內(nèi)存了，這里說(shuō)一下generator和函數(shù)的執(zhí)行流程，函數(shù)是順序執(zhí)行的，遇到return語(yǔ)句或者最后一行函數(shù)語(yǔ)句就返回。而變成generator的函數(shù)，在每次調(diào)用next()的時(shí)候執(zhí)行，遇到y(tǒng)ield語(yǔ)句返回，再次被next()調(diào)用時(shí)候從上次的返回yield語(yǔ)句處急需執(zhí)行，也就是用多少，取多少，不占內(nèi)存。

deffib(max):

n,a,b=0,0,1

whilen

yieldb

a,b=b,a+b

n=n+1

return'done'

a=fib(10)

print(fib(10))

print(a.__next__())

print(a.__next__())

print(a.__next__())

print("可以順便干其他事情")

print(a.__next__())

print(a.__next__())

結(jié)果：

可以順便干其他事情

在上面fib的例子，我們?cè)谘h(huán)過(guò)程中不斷調(diào)用yield，就會(huì)不斷中斷。當(dāng)然要給循環(huán)設(shè)置一個(gè)條件來(lái)退出循環(huán)，不然就會(huì)產(chǎn)生一個(gè)無(wú)限數(shù)列出來(lái)。同樣的，把函數(shù)改成generator后，我們基本上從來(lái)不會(huì)用next()來(lái)獲取下一個(gè)返回值，而是直接使用for循環(huán)來(lái)迭代：

deffib(max):

n,a,b=0,0,1

whilen

yieldb

a,b=b,a+b

n=n+1

return'done'

foriinfib(6):

print(i)

結(jié)果：

但是用for循環(huán)調(diào)用generator時(shí)，發(fā)現(xiàn)拿不到generator的return語(yǔ)句的返回值。如果拿不到返回值，那么就會(huì)報(bào)錯(cuò)，所以為了不讓報(bào)錯(cuò)，就要進(jìn)行異常處理，拿到返回值，如果想要拿到返回值，必須捕獲StopIteration錯(cuò)誤，返回值包含在StopIteration的value中：

deffib(max):

n,a,b=0,0,1

whilen

yieldb

a,b=b,a+b

n=n+1

return'done'

g=fib(6)

whileTrue:

try:

x=next(g)

print('generator:',x)

exceptStopIterationase:

print("生成器返回值：",e.value)

break

結(jié)果：

generator:1

generator:1

generator:2

generator:3

generator:5

generator:8

生成器返回值：done

還可以通過(guò)yield實(shí)現(xiàn)在單線程的情況下實(shí)現(xiàn)并發(fā)運(yùn)算的效果

由上面的例子我么可以發(fā)現(xiàn)，python提供了兩種基本的方式

生成器函數(shù)：也是用def定義的，利用關(guān)鍵字yield一次性返回一個(gè)結(jié)果，阻塞，重新開(kāi)始

生成器表達(dá)式：返回一個(gè)對(duì)象，這個(gè)對(duì)象只有在需要的時(shí)候才產(chǎn)生結(jié)果

以上內(nèi)容為大家介紹了什么是python生成器?，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。