[Python学习笔记-005] 理解yield

网络上介绍yield的文章很多，但大多讲得过于复杂或者追求全面以至于反而不好理解。本文用一个极简的例子给出参考资料[1]中的讲解，因为个人觉得其讲解最为通俗易懂，读者只需要对Python的列表有所了解即可。

When you see a function with yield statements, apply this easy trick to
understand what will happen:

1. Insert a line result = [] at the start of the function.
2. Replace each yield expr with result.append(expr).
3. Insert a line return result at the bottom of the function.
4. Yay - no more yield statements! Read and figure out code.
5. Compare function to original definition.

例如：

• foo1.py

 #!/usr/bin/python3
 import sys

 def foo(n):
     for i in range(n):
         yield i * i

 def main(argc, argv):
     for i in foo(int(argv[1])):
         print(i)
     return 0

 if __name__ == '__main__':
     sys.exit(main(len(sys.argv), sys.argv))

• foo2.py

 #!/usr/bin/python3
 import sys

 def foo(n):
     result = []
     for i in range(n):
         result.append(i * i)
     return result

 def main(argc, argv):
     for i in foo(int(argv[1])):
         print(i)
     return 0

 if __name__ == '__main__':
     sys.exit(main(len(sys.argv), sys.argv))

• meld foo1.py foo2.py

• 运行foo1.py 和 foo2.py

lijin$ ./foo1.py 

lijin$ ./foo2.py 

那么，问题来了，

Q1 - yield的含义是什么？

A：相当于return, 不过返回的是一个生成器函数。

Q2 - 为什么要引入yield?

A： 省内存！如果使用return一个list, 则list的元素全部储存在内存里；如果使用yield, 那就只需要在访问某个元素时将对应的元素装入内存即可。更多内容，请参考《Python学习手册》第4版第20章。

参考资料：

1. What does the "yield" keyword do?