python生成器yield和send

創建生成器

#第一種創建生成器的方法
#使用()創建生成器，如果使用[]則創建列表
a = (x**2 for x in range(1, 5))
#可以通過next一直產生新的數據，直到最后一個報異常，通過for遍歷不會報異常
#也可以使用a.__next__()
print(next(a))#輸出1
print(a.__next__())#輸出4
print(next(a))#輸出9

#通過for遍歷生成器
for i in a:
    print(i)

#第二種創建生成器的方法
def createNum(n):
    for i in range(n):
        yield i**3

for i in createNum(5):
    print(i)

 

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

#生成器的send用法 generator.send(value)
def test():
    i = 1
    while i < 5:
        temp = yield i**2
        print(temp)
        i += 1

t = test()
#第一次運行只能使用next或者send(None)
print(t.__next__())
#send的作用相當於使生成器繼續運行，並且傳遞的參數為yield的返回值(程序中即temp的值)
print(t.send("Hello World"))
print(t.__next__())#相當於send(None) 此時temp = None