Python迭代器和生成器

1、List Conprehensions

列表生成式是python中內(nèi)置的用來創(chuàng)建list的表達(dá)式、
如果我們要生成簡(jiǎn)單的list可以直接使用range函數(shù)來完成。

>>> range(10)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>

但是有時(shí)候我們要生成復(fù)雜的list怎么辦呢?比如我們生成一個(gè)[1^2, 2^2, 3^2...]的list。

>>> my_list = []
>>> for item in range(10):
...     my_list.append(item * item)
...
>>> my_list
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>>

上面的代碼都不夠簡(jiǎn)潔,python中的列表生成式可以一行解決。

>>> [x * x for x in range(10)]
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>>

在書寫生成式表達(dá)式時(shí),將要生成的元素放在前面,后面是for循環(huán)。
還可以在創(chuàng)建列表的同時(shí)進(jìn)行簡(jiǎn)單的過濾。

>>> [x * x for x in range(10) if x%2 == 0]
[0, 4, 16, 36, 64]
>>>

還可以實(shí)現(xiàn)全排列

>>> [(x,y) for x in '123' for y in '456']
[('1', '4'), ('1', '5'), ('1', '6'), ('2', '4'), ('2', '5'), ('2', '6'), ('3', '4'), ('3', '5'), ('3', '6')]
>>>

2、Iterables

當(dāng)我們創(chuàng)建了一個(gè)列表,我們可以依次遍歷它,依次遍歷的過程就是迭代。

>>> my_list = [1, 2, 3]
>>> for i in my_list:
...     print i
...
1
2
3

my_list就是一個(gè)可迭代對(duì)象,當(dāng)我們使用上面說的列表生成式創(chuàng)建一個(gè)列表時(shí),該列表也是一個(gè)可迭代的。

>>> my_list = [x * x for x in range(3)]
>>> for i in my_list:
...     print i
...
0
1
4

對(duì)于任何可迭代的對(duì)象都可以使用for in來進(jìn)行迭代。
但是這樣做有一個(gè)缺點(diǎn)就是:當(dāng)我們的數(shù)據(jù)非常多的時(shí)候比如一個(gè)大列表,這個(gè)列表將會(huì)被讀到內(nèi)存中,會(huì)占用非常多的內(nèi)存,有時(shí)候硬件是達(dá)不到的。

3、 Generators

對(duì)一個(gè)包含大量數(shù)據(jù)的列表,不僅占用很大的內(nèi)存,如果我們使用到的就只有某一小部分?jǐn)?shù)據(jù)就會(huì)導(dǎo)致內(nèi)存的浪費(fèi)。
如果列表中的元素可以根據(jù)某種算法推導(dǎo)出來,而不是將所有元素都存放在列表中,從而可以節(jié)省很多內(nèi)存空間。這樣一邊循環(huán)一邊計(jì)算的機(jī)制叫做生成器(Generator)。
創(chuàng)建一個(gè)Generator直接可以將上面的列表生成式的[]換成()就可以了。python中提供了生成器表達(dá)式,它是對(duì)列表和生成器的一種泛化,生成器表達(dá)式在運(yùn)行的時(shí)候,并不會(huì)將整個(gè)輸出序列都呈現(xiàn)出來,而是會(huì)估值為迭代器,這個(gè)迭代器每次可以根據(jù)生成器表達(dá)式產(chǎn)生一項(xiàng)數(shù)據(jù)。

>>> L1 = [x * x for x in range(10)]
>>> L2 = (x * x for x in range(10))
>>> L1
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> L2
<generator object <genexpr> at 0x000000000347F7E0>
>>>

從打印結(jié)果可以看出L2是一個(gè)Generator,可以使用next()方法打印每一個(gè)元素。

>>> L2.next()
0
>>> L2.next()
1
>>> L2.next()
4
>>> L2.next()
9
>>> L2.next()
16
>>> L2.next()
....
>>> L2.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>>

如果沒有元素了就會(huì)拋出異常。
Generator也是可迭代的,但是只能迭代一次。

>>> L2 = (x * x for x in range(10))
>>> for i in L2:
...     print i
...
0
1
4
9
16
25
36
49
64
81
>>>

4、生成器和迭代器區(qū)別

迭代器是一個(gè)更普遍的概念,如果一個(gè)類中有next()方法和返回自身return self__iter__()方法,這個(gè)類的對(duì)象就是一個(gè)Iterator
所有的生成器是一個(gè)迭代器,但是反過來就不是。生成器通過調(diào)用具有一個(gè)或多個(gè)yield表達(dá)式(在Python 2.5和更早版本中的yield語句)的函數(shù)構(gòu)建。

>>> def squares(start, stop):
...     for i in range(start, stop):
...             yield i * i
...
>>>
>>> generator = squares(1, 3)
>>> generator
<generator object squares at 0x000000000347F7E0>
>>>

當(dāng)然還可以使用生成器表達(dá)式直接生成一個(gè)Generator。
但是有時(shí)候你需要一個(gè)定制的迭代器,定義一個(gè)類,該類實(shí)現(xiàn)next() 和 __iter__()方法就可以。

class Squares(object):
    def __init__(self, start, stop):
        self.start = start
        self.stop = stop
    def __iter__(self):
        return self
    def next(self):
        if self.start >= self.stop:
            raise StopIteration
        current = self.start * self.start
        self.start += 1
        return current
5、使用yield關(guān)鍵字創(chuàng)建生成器

如果一個(gè)函數(shù)包含了yield關(guān)鍵字,這個(gè)函數(shù)就是一個(gè)生成器。但是Generator和函數(shù)的執(zhí)行流程是不一樣的。函數(shù)是順序執(zhí)行,遇到return語句或者執(zhí)行完畢后就返回,如果變成了Generator后,每次調(diào)用next()函數(shù)開始執(zhí)行,遇到yield返回,下次接著從上次返回位置進(jìn)行執(zhí)行。

>>> def fun():
...     print 'run 1'
...     yield 1
...     print 'run 2'
...     yield 2
...
>>> g = fun()
>>> g.next()
run 1
1
>>> g.next()
run 2
2
>>> g.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>>

我們使用關(guān)鍵字yield定義了一個(gè)Generator,在執(zhí)行的時(shí)候,遇到yield關(guān)鍵字就會(huì)返回,再次調(diào)用就從上次返回的地方開始。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,406評(píng)論 6 538
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 99,034評(píng)論 3 423
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 177,413評(píng)論 0 382
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我,道長(zhǎng),這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,449評(píng)論 1 316
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 72,165評(píng)論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,559評(píng)論 1 325
  • 那天,我揣著相機(jī)與錄音,去河邊找鬼。 笑死,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,606評(píng)論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,781評(píng)論 0 289
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 49,327評(píng)論 1 335
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 41,084評(píng)論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 43,278評(píng)論 1 371
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,849評(píng)論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 44,495評(píng)論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,927評(píng)論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,172評(píng)論 1 291
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 52,010評(píng)論 3 396
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 48,241評(píng)論 2 375

推薦閱讀更多精彩內(nèi)容