溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

python yield和Generator函數(shù)用法詳解

發(fā)布時間:2020-09-18 14:31:04 來源:腳本之家 閱讀:169 作者:郭雪原 欄目:開發(fā)技術(shù)

這篇文章主要介紹了python yield和Generator函數(shù)用法詳解,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

首先我們從一個小程序?qū)?,各定一個list,找出其中的素數(shù),我們會這樣寫

import math
def is_Prims(number):

  if number == 2:
    return True
  //除2以外的所有偶數(shù)都不是素數(shù)
  elif number % 2 == 0:
    return False
  //如果一個數(shù)能被除1和本身之外的數(shù)整除,則為合數(shù)。其實我們的判定范圍到根號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)

但如果我們想給定一個數(shù),然后列出比這個數(shù)大的所有素數(shù)呢?我們可能這樣寫:

def get_Prims(number):
  if is_Prims(number):
    return number

但是一旦return函數(shù)將控制權(quán)交給調(diào)用者后徹底結(jié)束,任何局部變量和函數(shù)工作都被丟棄,下一次調(diào)用又會從頭開始。因此我們就可以用一下寫法:

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函數(shù),一個函數(shù)的def代碼里包含了yield,函數(shù)就自動成為了一個generator函數(shù)(及時仍包含有return),generator函數(shù)創(chuàng)建generator(一種特殊形式的迭代器,這個迭代器有一個內(nèi)置__next__()方法),當(dāng)需要一個值的時候通過yield來產(chǎn)生而不是直接return,因此與一般函數(shù)不同的是,此時控制權(quán)并未交出。

for循環(huán)會隱式的調(diào)用next()函數(shù),next()函數(shù)負責(zé)調(diào)用generator中的__next__()方法,此時generator負責(zé)返回一個值給任何調(diào)用next()的方法,利用yield將此值傳回去,相當(dāng)于return語句。

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。

向AI問一下細節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI