快捷導(dǎo)航

Python中yield返回生成器的詳細方法

更新時間：2021年11月03日 15:54:50 作者：python學(xué)習(xí)者0

這篇文章主要介紹了Python中的yield返回生成器,生成器是Python編程進階中的重要知識點,需要的朋友可以參考下，希望能夠給你帶來幫助

一、迭代器

逐項讀取列表，稱為迭代。

mylist = [1, 2, 3]
for i in mylist: # 可迭代對象
    print(i)

列表解析式同樣是一個迭代器。

mylist = [x*x for x in range(3)]
for i in mylist:
    print(i)
'''
0
1
4
'''

所有 for...in... 都是迭代器，包括列表、字符串、文件等等。

但是，迭代器所有的值都存儲在內(nèi)存中，十分浪費內(nèi)存。

因此有了生成器的概念。

二、生成器

生成器是一種迭代器，這種迭代器只能迭代一次。

生成器不會一次性存儲所有的值，而是會動態(tài)的生成值。

mygenerator = (x*x for x in range(3))
for i in mygenerator:
    print(i)

生成器只可執(zhí)行一次，再次執(zhí)行時不會輸出任何東西。

三、yield

1.例子一

yield 類似于 return 關(guān)鍵字，只是函數(shù)將返回一個生成器。

# 創(chuàng)建生成器
def createGenerator():
    mylist = range(10)
    for i in mylist:
        print(i) # 驗證函數(shù)調(diào)用時并無執(zhí)行
        yield i*i
mygenerator = createGenerator()    
print(mygenerator) 
# <generator object createGenerator at 0x0000029E88FDCA50>
# 使用生成器
for i in mygenerator:
    print(i)
# 再次執(zhí)行 返回為空 沒有值了

函數(shù)將返回一組只需要讀取一次的值，可以大大的提升代碼性能。

在調(diào)用函數(shù)時，函數(shù)體中的代碼并不會執(zhí)行，函數(shù)只返回生成器對象。

代碼每次從使用生成器時停止的地方繼續(xù)。

2.例子二

#Python學(xué)習(xí)交流群：531509025
# 學(xué)習(xí)另外一個例子
def foo():
    print("starting...")
    while True:
        res = yield 4 # 函數(shù)并不真正執(zhí)行
        print("res:", res)
g = foo() # 得到一個生成器對象
print(next(g)) # 真正執(zhí)行
print("*"*20)
print(next(g)) # 從上一次停止的地方繼續(xù)執(zhí)行
'''
starting...
4
********************
res: None
4
'''
print(g.send(7))

執(zhí)行 yield 之后，才會跳出 while 循環(huán)。

next 函數(shù)用于執(zhí)行下一步操作。

send 函數(shù)用于發(fā)送一個參數(shù)給生成器。且 send 方法中包含 next 方法。