python生成器(筆記)

Python生成器有些難以用語言表達其概念,所以在這裏用幾段代碼來解釋~

生成器:任何包含yield語句的函數稱爲生成器;

              生成器是一種普通的函數語法定義的迭代器。


def test2():
    print 9
    print 8
    yield 7
test2()

上述代碼什麼都不返回,因爲代碼碰到了yield函數暫停(或凍結)了,這個暫停同時還影響了yield以上的兩個print。


def test2():
    print 9
    print 8
    yield 7
t2 = test2()
t2.next()
--------------------------------------------------------------------------------
[root@xfwy tmp]# ./1.py
9
8

通過使用next方法,激活了yield,這樣被yield暫停的兩條print語句得以輸出。

但是,yield後的7是什麼?看樣子是想要輸出的,但是怎麼沒有輸出?

這是因爲,使用next方法,只是讓凍結內容繼續執行,而yield只是激活了,而它自身並不會輸出它後邊的值,而是保存下來。


那這裏yield已經將值保存出來了,怎麼輸出呢~

def test2():
    print 9
    print 8
    yield 7
t2 = test2()
print t2.next()
--------------------------------------------------------------------------------
[root@xfwy tmp]# ./1.py
9
8
7

對的,只需要一個print,以上兩段代碼就可以看出來,t2.next()只是將暫停繼續(激活),並保存yield的值,而yield語句並不會輸出值,這樣就得使用print來打印出值。

這樣,大致的yield的概念清楚了,我們就理解下下邊這個二層嵌套列表的生成器了

def flt(nes):
    for i in nes:
        for j in i:
            yield j
nested = [[1, 2], [3, 4], [5]]
for num in flt(nested):
    print num
print "---------"
print list(flt(nested))
--------------------------------------------------------------------------------
[root@xfwy tmp]# ./1.py   
1
2
3
4
5
---------
[1, 2, 3, 4, 5]

因爲生成器有個概念爲“生成器是一種普通的函數語法定義的迭代器”。所以生成器就是迭代器且可被迭代。

函數flt()的兩層for循環很清楚,就是從一個二層嵌套表中取元素。上述代碼是用迭代for循環取元素,取一個元素,操作一個元素。

所以,yield作用就是,yield語句在的話,yield語句是整個函數的主權,所有代碼不管你yield代碼前返回什麼或者打印什麼,都會被暫停,直到有激活出現。


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章