python yield和Generator函數用法詳解
這篇文章主要介紹了python yield和Generator函數用法詳解,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
首先我們從一個小程序導入,各定一個list,找出其中的素數,我們會這樣寫
import math
def is_Prims(number):
if number == 2:
return True
//除2以外的所有偶數都不是素數
elif number % 2 == 0:
return False
//如果一個數能被除1和本身之外的數整除,則為合數。其實我們的判定范圍到根號n就可以
for cur in range(2,int(math.sqrt(number))+1,2):
if number % cur == 0:
return False
else:
return True
def get_Prims(input_list):
result_list = list()
for element in input_list:
if is_Prims(element):
result_list.append(element)
return result_list
aa = get_Prims([1,2,3,4,5,6,7,8,9])
print (aa)
但如果我們想給定一個數,然后列出比這個數大的所有素數呢?我們可能這樣寫:
def get_Prims(number):
if is_Prims(number):
return number
但是一旦return函數將控制權交給調用者后徹底結束,任何局部變量和函數工作都被丟棄,下一次調用又會從頭開始。因此我們就可以用一下寫法:
def get_Prims(number):
while(True):
if is_Prims(number):
yield number
number += 1
def get_numbers():
total = list()
for next_prim in get_Prims(2):
if next_prim < 100:
total.append(next_prim)
else:
print(total)
return
get_numbers()
下面解釋一下generator函數,一個函數的def代碼里包含了yield,函數就自動成為了一個generator函數(及時仍包含有return),generator函數創(chuàng)建generator(一種特殊形式的迭代器,這個迭代器有一個內置__next__()方法),當需要一個值的時候通過yield來產生而不是直接return,因此與一般函數不同的是,此時控制權并未交出。
for循環(huán)會隱式的調用next()函數,next()函數負責調用generator中的__next__()方法,此時generator負責返回一個值給任何調用next()的方法,利用yield將此值傳回去,相當于return語句。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
django數據模型中null和blank的區(qū)別說明
這篇文章主要介紹了django數據模型中null和blank的區(qū)別說明,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-09-09
python項目中requirements.txt的用法實例教程
Python項目中必須包含一個requirements.txt文件,用于記錄所有依賴包及其精確的版本號,以便新環(huán)境部署,下面這篇文章主要給大家介紹了關于python項目中requirements.txt用法的相關資料,需要的朋友可以參考下2022-06-06
Pytorch技巧:DataLoader的collate_fn參數使用詳解
今天小編就為大家分享一篇Pytorch技巧:DataLoader的collate_fn參數使用詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01

