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函數創建generator（一種特殊形式的迭代器，這個迭代器有一個內建__next__()方法），當需要一個值的時候透過yield來產生而不是直接return，因此與一般函數不同的是，此時控制權並未交出。

for迴圈會隱式的呼叫next()函數，next()函數負責呼叫generator中的__next__()方法，此時generator負責傳回一個值給任何呼叫next()的方法，利用yield將此值傳回去，相當於return語句。

更多python之yield和Generator分析相關文章請關注PHP中文網！