python生成器yield和send

建立生成器

 1 #第一種建立生成器的方法
 2 #使用()建立生成器，若是使用[]則建立列表
 3 a = (x**2 for x in range(1, 5))  4 #能夠經過next一直產生新的數據，直到最後一個報異常，經過for遍歷不會報異常
 5 #也可使用a.__next__()
 6 print(next(a))#輸出1
 7 print(a.__next__())#輸出4
 8 print(next(a))#輸出9
 9 
10 #經過for遍歷生成器
11 for i in a: 12     print(i) 13 
14 #第二種建立生成器的方法
15 def createNum(n): 16     for i in range(n): 17         yield i**3
18 
19 for i in createNum(5): 20     print(i)

 

生成器的send用法 generator.send(value)

 1 #生成器的send用法 generator.send(value)
 2 def test():  3     i = 1
 4     while i < 5:  5         temp = yield i**2
 6         print(temp)  7         i += 1
 8 
 9 t = test() 10 #第一次運行只能使用next或者send(None)
11 print(t.__next__()) 12 #send的做用至關於使生成器繼續運行，而且傳遞的參數爲yield的返回值(程序中即temp的值)
13 print(t.send("Hello World")) 14 print(t.__next__())#至關於send(None) 此時temp = None