快捷導(dǎo)航

Python 生成器，迭代，yield關(guān)鍵字，send()傳參給yield語句操作示例

更新時間：2019年10月12日 11:18:47 作者：houyanhua1

這篇文章主要介紹了Python 生成器，迭代，yield關(guān)鍵字，send()傳參給yield語句操作,結(jié)合實例形式分析了Python生成器、迭代、yield關(guān)鍵字及異常處理相關(guān)操作技巧,需要的朋友可以參考下

本文實例講述了Python 生成器，迭代，yield關(guān)鍵字，send()傳參給yield語句操作。分享給大家供大家參考，具體如下：

demo.py（生成器，yield關(guān)鍵字）：

# 生成器是一個特殊的迭代器?？梢杂胒or...in遍歷。
# 帶有yield關(guān)鍵字的函數(shù),不再是一個函數(shù),而是一個生成器模板。調(diào)用該模板會返回一個生成器對象。
def create_num(all_num):
  a, b = 0, 1
  current_num = 0
  while current_num < all_num:
    yield a  # 當(dāng)遍歷create_num返回的生成器時,會阻塞在yield的位置。每次遍歷出的值都是yield后的值。
    a, b = b, a+b
    current_num += 1
  # return '返回值'  # 迭代結(jié)束后,繼續(xù)調(diào)用next會拋StopIteration異常。 可以通過該異常來獲取該返回值。 (異常.value 就是該返回值)
# create_num"函數(shù)"中有一個yield,那么create_num不再是一個函數(shù)。調(diào)用時，會返回一個生成器對象。
obj = create_num(10) # 只會返回一個生成器對象(可用于遍歷)。并不會執(zhí)行create_num中的代碼,只有遍歷(迭代)時才會執(zhí)行create_num中的代碼。
ret = next(obj) # 迭代的本質(zhì)就是調(diào)用對象的__next__函數(shù)。 會返回yield后面的值,并阻塞代碼,直到再次調(diào)用next(或迭代)才會解阻塞。
print(ret)  # 當(dāng)create_num中的代碼執(zhí)行完后，迭代就會結(jié)束。
ret = next(obj) # 如果迭代結(jié)束后,繼續(xù)調(diào)用next,那么會拋異常。 可以通過異常來獲取create_num return的值。
print(ret)  # 可以通過異常來判斷是否迭代結(jié)束。
obj2 = create_num(2) # obj2和obj的遍歷迭代互不影響。
ret = next(obj2)
print(ret)
# for num in obj:
#  print(num)

運行結(jié)果：

0
1
0

demo.py（通過異常判斷迭代是否結(jié)束）：

def create_num(all_num):
  a, b = 0, 1
  current_num = 0
  while current_num < all_num:
    yield a 
    a, b = b, a+b
    current_num += 1
  return "ok...."  # 通過迭代結(jié)束后的異常來獲取該返回值
obj = create_num(10) # 返回一個生成器對象。并不會執(zhí)行create_num中的代碼，只有遍歷迭代obj時才會執(zhí)行create_num中的代碼
while True:
  try:
    ret = next(obj) # 迭代結(jié)束后繼續(xù)調(diào)用next會拋異常。
    print(ret)
  except Exception as ret:
    print(ret.value) # 通過異常獲取create_num return的值。
    break

運行結(jié)果：

0
1
1
2
3
5
8
13
21
34
ok....

demo.py（send()迭代生成器，傳參給yield語句）：

def create_num(all_num):
  a, b = 0, 1
  current_num = 0
  while current_num < all_num:
    ret = yield a  # send的參數(shù)就是yield語句的返回值。
    print(">>>ret>>>>", ret)  # hahahha
    a, b = b, a+b
    current_num += 1
obj = create_num(10)
# obj.send(None) # send一般不會放到第一次啟動(迭代)生成器，如果非要這樣做 那么傳遞None (否則會拋異常)
ret = next(obj) # 第一次遍歷迭代生成器時，建議使用next函數(shù)。
print(ret)
# send與next作用相同，都是進(jìn)行下一次迭代的意思。 (都會解阻塞yield關(guān)鍵字)
# send可以傳遞參數(shù)表示yield語句的返回值。 而next不能傳遞參數(shù)。
ret = obj.send("hahahha") # 會先將"hahahha"參數(shù)當(dāng)做yield語句的返回值，然后再解阻塞yield 遍歷。(因此不推薦第一次遍歷時使用send傳參) 
print(ret)

運行結(jié)果：