python 中 yield 的使用
python中yield在函數(shù)中的使用
正是因?yàn)楹瘮?shù)含有這個(gè)yield,所以,該函數(shù)不再是普通的函數(shù),而是生成器函數(shù)(generator function)。下面通過(guò)小例子來(lái)說(shuō)明一下這個(gè)內(nèi)置函數(shù)的特性:
1、直接運(yùn)行函數(shù)并不會(huì)執(zhí)行。
a = 0 def add_(initial, k):for i in range(k):initial = initial + 1print(initial)yield initial add_(a, 10)從運(yùn)行結(jié)果發(fā)現(xiàn),并沒(méi)有輸出initial的值,說(shuō)明該函數(shù)并沒(méi)有直接調(diào)用執(zhí)行。
2、直接打印該函數(shù)會(huì)返回一個(gè)生成器對(duì)象。
print(add_(a, 10)) <generator object add_ at 0x000002457504CBA0>3、可通過(guò)for循環(huán)迭代執(zhí)行。
a = 0 def add_(initial, k):for i in range(k):initial = initial + 1print(initial)yield initial add_(a, 4) print(add_(a, 4)) for j in add_(a, 4):print('j:', j) <generator object add_ at 0x000002A92F81CBA0> 1 j: 1 2 j: 2 3 j: 3 4 j: 4Process finished with exit code 0從輸出結(jié)果可以得知,每次函數(shù)執(zhí)行到y(tǒng)ield函數(shù)時(shí)會(huì)終止函數(shù)的執(zhí)行,并保留變量initial的值,通過(guò)for循環(huán)再次執(zhí)行該函數(shù)時(shí),會(huì)從上次執(zhí)行到的地方跳過(guò)yield繼續(xù)執(zhí)行。
總結(jié):yield 的作用就是把一個(gè)函數(shù)變成一個(gè) generator,帶有 yield 的函數(shù)不再是一個(gè)普通函數(shù),Python 解釋器會(huì)將其視為一個(gè) generator,調(diào)用 add_(a, 4)不會(huì)執(zhí)行add_函數(shù),而是返回一個(gè) iterable 對(duì)象。在 for 循環(huán)執(zhí)行時(shí),每次循環(huán)都會(huì)執(zhí)行 add_函數(shù)內(nèi)部的代碼,執(zhí)行到 yield initial?時(shí),add_函數(shù)就返回一個(gè)指定變量的迭代值,在示例中,這個(gè)指定的變量為initial,下次迭代時(shí),代碼從 yield initial的下一條語(yǔ)句繼續(xù)執(zhí)行,而函數(shù)的本地變量看起來(lái)和上次中斷執(zhí)行前是完全一樣的,于是函數(shù)繼續(xù)執(zhí)行,直到再次遇到 yield。
該方法可用于for循環(huán)迭代獲取數(shù)據(jù),為模型本身節(jié)省了內(nèi)存空間與運(yùn)算時(shí)間。
總結(jié)
以上是生活随笔為你收集整理的python 中 yield 的使用的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: pytorch数据加载时报错OSErro
- 下一篇: python+OpenCV检测条形码