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語句。

 

參考：http://article.yeeyan.org/view/75219/422747