yield的问题

Question

>>> def func(n):
...     for i in range(0,n):
...         arg= yield i
...         print(i,arg)
... 
>>> f=func(3)
>>> while True:
...     print('main',next(f))
... 
main 0
0 None
main 1
1 None
main 2
2 None

为何结果不是

main 0
0 0
main 1
1 1
main 2
2 2

Answer 1

需要正确使用send方法~

def func(n):
     for i in range(0,n):
        arg= yield i
        print(i,arg)

f = func(3)
f.send(None)# or f.next()
i = 0
while True:
    try:
       rc = f.send(i)
       print('main', rc)
       i += 1
    except StopIteration:
        print('Done')
        break

Answer 2

arg= yield i这句里
arg是 send 过去的。
你在启动f后，使用 f.send(somevalue), 你就可以看见arg就是somevalue了。

Answer 3

如果需要返回请使用send通信

一楼的正解，要不然你只能获取yeild得返回值

Answer 4

你的func根本没有return语句，当然是None,应该这么使用：

for i in func():
    print('main',i)