快捷導(dǎo)航

Python中Generators教程的實(shí)現(xiàn)

更新時(shí)間：2023年02月14日 14:28:17 作者：代碼輸入中...

本文主要介紹了Python中Generators教程的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

怎樣創(chuàng)建一個(gè)python generator？

就像創(chuàng)建一個(gè)函數(shù)一樣簡(jiǎn)單，只不過(guò)不使用return 聲明，而是使用yield聲明.

如果一個(gè)函數(shù)至少包含一個(gè)yield聲明(當(dāng)然它也可以包含其他yield或return)，那么它就是一個(gè)generator.

yield和return都會(huì)讓函數(shù)返回一些東西，區(qū)別在于，return聲明徹底結(jié)束一個(gè)函數(shù)，而yield聲明是暫停函數(shù)，保存它的所有狀態(tài)，并且后續(xù)被調(diào)用后會(huì)繼續(xù)執(zhí)行.

generator函數(shù)和普通函數(shù)的區(qū)別

generator函數(shù)包含一個(gè)以上的yield聲明
generator函數(shù)被調(diào)用的時(shí)候，會(huì)返回一個(gè)iterator對(duì)象，但是函數(shù)并不會(huì)立即開始執(zhí)行
__iter__()和__next__()方法被自動(dòng)實(shí)現(xiàn)，所以可以使用next()函數(shù)對(duì)返回的此iterator對(duì)象進(jìn)行迭代
一旦一個(gè)generator 執(zhí)行到y(tǒng)ield語(yǔ)句，generator函數(shù)暫停，程序控制流被轉(zhuǎn)移到調(diào)用方
在對(duì)generator的連續(xù)調(diào)用之間，generator的本地變量和狀態(tài)會(huì)被保存
最終，generator函數(shù)終止，再調(diào)用generator會(huì)引發(fā)StopIteration異常

下面這個(gè)例子說(shuō)明上述全部要點(diǎn)，我們有一個(gè)名為my_gen()的函數(shù)，它帶有一些yield聲明.

# A simple generator function  
def my_gen():  
    n = 1  
    print('This is printed first')  
    # Generator function contains yield statements  
    yield n  
  
    n += 1  
    print('This is printed second')  
    yield n  
  
    n += 1  
    print('This is printed at last')  
    yield n

有趣的是，在這個(gè)例子里變量n在每次調(diào)用之間都被記住了。和一般函數(shù)不同的是，在函數(shù)yield之后本地變量沒(méi)有被銷毀，而且，generator對(duì)象只能被這樣迭代一次。

要想重復(fù)上面的過(guò)程，需要類似 a = my_gen() 這樣創(chuàng)建另一個(gè)generator對(duì)象，并對(duì)其使用next方法迭代。

注意

：我們可以對(duì)generator對(duì)象直接使用for循環(huán)。

這是因?yàn)橐粋€(gè)for循環(huán)接收一個(gè)iterator對(duì)象，且使用next()函數(shù)迭代它，當(dāng)遇到StopIteration異常的時(shí)候自動(dòng)停止。

# A simple generator function  
def my_gen():  
    n = 1  
    print('This is printed first')  
    # Generator function contains yield statements  
    yield n  
  
    n += 1  
    print('This is printed second')  
    yield n  
  
    n += 1  
    print('This is printed at last')  
    yield n  
  
# Using for loop  
  
# Output:   
# This is printed first  
# 1  
# This is printed second  
# 2  
# This is printed at last  
# 3  
  
for item in my_gen():  
    print(item)

有循環(huán)的python generator

上面的例子沒(méi)有實(shí)際的應(yīng)用意義，我們只是為了探究背后原理。

通常來(lái)說(shuō)，generator都是和循環(huán)結(jié)合實(shí)現(xiàn)的，且這個(gè)循環(huán)帶有一個(gè)終止條件。

我們來(lái)看一個(gè)reverse一個(gè)字符串的例子

def rev_str(my_str):  
    length = len(my_str)  
    for i in range(length - 1,-1,-1):  
        yield my_str[i]  
  
# For loop to reverse the string  
# Output:  
# o  
# l  
# l  
# e  
# h  
for char in rev_str("hello"):  
     print(char)

我們?cè)趂or循環(huán)里面使用range()函數(shù)來(lái)獲取反向順序的index。

generator除了可以應(yīng)用于string，還可以應(yīng)用于其它類型的iterator，例如list，tuple等。

python generator 表達(dá)式

使用generator表達(dá)式可以很容易地創(chuàng)建簡(jiǎn)單的generator。

就像lambda函數(shù)可以創(chuàng)建匿名函數(shù)一樣，generator函數(shù)創(chuàng)建一個(gè)匿名generator函數(shù)。

generator表達(dá)式的語(yǔ)法類似于python的list comprehension，只是方括號(hào)被替換為了圓括號(hào)而已。

list comprehension和generator表達(dá)式的主要區(qū)別在于，前者產(chǎn)生全部的list，后者每次僅產(chǎn)生一項(xiàng)。

它們有些懶惰，僅在接到請(qǐng)求的時(shí)候才會(huì)產(chǎn)生輸出。因此，generator表達(dá)式比list comprehension更加節(jié)省內(nèi)存。

# Initialize the list  
my_list = [1, 3, 6, 10]  
  
# square each term using list comprehension  
# Output: [1, 9, 36, 100]  
[x**2 for x in my_list]  
  
# same thing can be done using generator expression  
# Output: <generator object <genexpr> at 0x0000000002EBDAF8>  
(x**2 for x in my_list)

上面的例子中，generator表達(dá)式?jīng)]有立即產(chǎn)生需要的結(jié)果，而是在需要產(chǎn)生item的時(shí)候返回一個(gè)generator對(duì)象。

# Intialize the list  
my_list = [1, 3, 6, 10]  
  
a = (x**2 for x in my_list)  
# Output: 1  
print(next(a))  
  
# Output: 9  
print(next(a))  
  
# Output: 36  
print(next(a))  
  
# Output: 100  
print(next(a))  
  
# Output: StopIteration  
next(a)

generator表達(dá)式可以在函數(shù)內(nèi)部使用。當(dāng)這樣使用的時(shí)候，圓括號(hào)可以丟棄。

python里為什么要使用generator？

1.容易實(shí)現(xiàn)

相對(duì)于iterator類來(lái)說(shuō)，generator的實(shí)現(xiàn)清晰、簡(jiǎn)潔。下面是用iterator實(shí)現(xiàn)一個(gè)2的指數(shù)函數(shù)

class PowTwo:  
    def __init__(self, max = 0):  
        self.max = max  
  
    def __iter__(self):  
        self.n = 0  
        return self  
  
    def __next__(self):  
        if self.n > self.max:  
            raise StopIteration  
  
        result = 2 ** self.n  
        self.n += 1  
        return result

generator這樣實(shí)現(xiàn)

def PowTwoGen(max = 0):  
    n = 0  
    while n < max:  
        yield 2 ** n  
        n += 1

因?yàn)間enerator自動(dòng)跟蹤實(shí)現(xiàn)細(xì)節(jié)，因此更加清晰、簡(jiǎn)潔。

2.節(jié)省內(nèi)存

一個(gè)函數(shù)返回一個(gè)序列(sequence)的時(shí)候，會(huì)在內(nèi)存里面把這個(gè)序列構(gòu)建好再返回。如果這個(gè)序列包含很多數(shù)據(jù)的話，就過(guò)猶不及了。

而如果序列是以generator方式實(shí)現(xiàn)的，就是內(nèi)存友好的，因?yàn)樗看沃划a(chǎn)生一個(gè)item。

3.代表無(wú)限的stream

generator是一個(gè)很棒的表示無(wú)限數(shù)據(jù)流的工具。無(wú)限數(shù)據(jù)流不能被保存在內(nèi)存里面，并且因?yàn)間enerator每次產(chǎn)生一個(gè)item，它就可以表示無(wú)限數(shù)據(jù)流。

下面的代碼可以產(chǎn)生所有的奇數(shù)

def all_even():  
    n = 0  
    while True:  
        yield n  
        n += 2

4.generator流水線(pipeline)

generator可以對(duì)一系列操作執(zhí)行流水線操作。

假設(shè)我們有一個(gè)快餐連鎖店的日志。日志的第四列是每小時(shí)售出的披薩數(shù)量，我們想對(duì)近5年的這一數(shù)據(jù)進(jìn)行求和。

假設(shè)所有數(shù)據(jù)都是字符，不可用的數(shù)據(jù)都以"N/A"表示，使用generator可以這樣實(shí)現(xiàn)

with open('sells.log') as file:  
    pizza_col = (line[3] for line in file)  
    per_hour = (int(x) for x in pizza_col if x != 'N/A')  
    print("Total pizzas sold = ",sum(per_hour))

這個(gè)流水線既高效又易讀，并且看起來(lái)很酷?。海?/p>

到此這篇關(guān)于Python中Generators教程的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Python Generators內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: