python高级进阶_8_使用生成器send(arg)注意事项03

看下下边的代码：

def test():
    i=0
    while i<5:
        temp=yield i*i
        print(temp)
        i=i+1

a=test()     # 生成器对象 ，里面都是存储 i*i的值

a.send("heihei")

但是报错了

    a.send("heihei")
TypeError: can't send non-None value to a just-started generator

请思考几分钟 ，它为什么会报错呢

---

看你想的和我一样吗 ？

大家知道 当运行yield的时候会停顿，然后保存值， 上边的代码 第一次运行的时候，就运行到
yield i*i 就不会继续运行了， 所以send 传入的参数没有任何会接受，所以要报错。如果不想报错，可以这么改

看下代码：

def test():
    i=0
    while i<5:
        temp=yield i*i
        print(temp)
        i=i+1

a=test()     # 生成器对象 ，里面都是存储 i*i的值

print(a.__next__())   #  这一步 会运行到 yield i*i，
print(a.send("heihei"))  #  这一步 会继续运行下边的代码， temp= yield  接受传来的参数。  然后再打印，直到再次yield 保存 i*i

打印结果：

0
heihei
1

有些小伙伴该问了，我就想用send ，不想第一次用next，怎么解决呢?

修改如下，自己尝试下 运行结果和上边的一样;

def test():
    i=0
    while i<5:
        temp=yield i*i
        print(temp)
        i=i+1

a=test()     # 生成器对象 ，里面都是存储 i*i的值

print(a.send(None))  # 这个不传入参数， 也就和 next（） 用法是一样的。
print(a.send("heihei"))