python中的yield通俗理解

I、理解yield，第一步需要理解yield和return的區別

print('yield:')
def _testyield():
    for i in range(5):
        yield i*i
#這里產生生成器對象,跟java對象意思相同
generator = _testyield()

for i in range(5):
    print(next(generator))

#-----------------------------------------------------------#

print('return:')
def _return(n):
    # 這里res是一個list[],得出的結果是[1,2,3,4,5]
    res = [i*i for i in range(n)]    
    return res

for i in _return(5):
    print(i)

上面yield和return生成的結果相同：

但是這里面的區別在於：

return返回的是一個list列表，而yield每次調用只返回一個數值，毫無疑問，使用return空間開銷比較大，尤其是操作巨量數據的時候，操作一個大列表時間開銷也會得不償失

 

II、yield執行方式

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)


g = foo()
#此時未打印出任何信息，說明只是生成了generator對象，並未執行函數

print('-'*30)
print('g_1',next(g))
#調用next，開始執行函數，並且到達yield這一步時，返回生成的4

print("*"*20)
print('g_2',next(g))
#res：none，說明是接着上一步開始執行，並且第二次循環到達yield這一步

print("#"*30)
print('g_3',next(g))

打印結果如下：

 

 說明：

1、調用包含yield函數時，並不會執行函數，而是產生並返回一個生成器對象

2、第一次next取出一個值時，會將函數執行到第一個yield，停止

3、后面next時，會從上一個yield開始，接着執行，循環到下一個yield

后面均參照第三行。

 

III、繼續深挖yield

def foo():
    print("starting...")
    for i in range(6):
        res = yield 4
        print("res:",res)

    #當我們添加新的yield
    for i in range(6):
        res = yield 5
        print("res",res)

g = foo()
#此時未打印出任何信息，說明只是生成了generator對象，並未執行函數

for i in range(7):
    print(next(g))

結果是：

 

 上述結果5最后出現，說明yield會按照正常順序進行執行，不會重新加載函數