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