python yield和Generator函數用法詳解

更新時間：2020年02月10日 09:17:44 作者：郭雪原

這篇文章主要介紹了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語句。

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: