如何在python3.6項目中使用yield生成器

這期內容當中小編將會給大家帶來有關如何在python3.6項目中使用yield生成器，文章內容豐富且以專業的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

定義一個函數test()，看看return的效果

def test(n):
  for i in range(n):
    return i
    print('mark')

測試結果

for i in range(5):
  print(test(3))

輸出

0
0
0
0
0

每次調用函數時碰到return就自動結束了，返回當前i值

再來看看yield的效果：

def test2(n):
  for i in range(n):
    yield i
    print('mark')

測試

for i in range(5):
  print(test2(3))

輸出

<generator object test2 at 0x12d245200>
<generator object test2 at 0x12d245200>
<generator object test2 at 0x12d245200>

只要函數中含有yield，python就會默認這個函數是一個生成器，這個測試相當于重復打開了三次生成器

生成器需要用next()調用

測試

t = test2(3)
for i in range(3):
  print('i=',i)
  print(next(t))

next(t) 等同于 t.__next__()

結果

i= 0
0
i= 1
mark
1
i= 2
mark
2

可以看到，第一次返回i=0之后，第二次再調用函數時，從yield i的下一步繼續，即print('mark')。

生成器還支持用send()將值傳遞進函數：

def test3():
  value = (yield 1)
  print(value)
  print('mark1')
  value = (yield value)
  print(value)
  print('mark2')

測試

t2 = test3()
print(t2.__next__())
print('-----------')
print(t2.send(2))
print('-----------')
print(t2.send(3))

輸出

1
-----------
2
mark1
2
-----------
3
mark2
---------------------------------------------------------------------------
StopIteration                             Traceback (most recent call last)
<ipython-input-102-56e62df29d71> in <module>()
      4 print(t2.send(2))
      5 print('-----------')
----> 6 print(t2.send(3))

StopIteration:

第一次next()返回1，并在當前停止，send(2)會將2傳輸給當前停止的那一步yield處，即相當于value = (yield 1)變為value = (2)

然后繼續運行

print(value)
print('mark1')
value = (yield value)

碰到yield再次停止，返回此時的vlue值2，之后send(3)將3傳輸給當前停止那一步yield處，即第二個yield value處，繼續運行

print(value)
print('mark2')

上述就是小編為大家分享的如何在python3.6項目中使用yield生成器了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業資訊頻道。