yield 介绍以及运用示例【python 3】

【摘要】在 python 3 中 yield 运用比较广泛，但是这个 yield 概念是不好理解的，我也是看了些许文档，运行了若干程序后才领悟出来了，本文就关于 yield 概念在自己的理解下通过代码示例进行概括、总结。

在 python 3 中任何使用 yield 的函数都称之为生成器。以下的话，直接给出示例，大家通过多动手去慢慢的理解。

注意：yield i 含义和 return i 相似。

例子 1 ：

1. # 生成器实际上是每一次都在调用 .__next__() , 每调用一次 .__next__(), 就返回 yield 后面的值，等价于 return

3. def test():

4.     print('first')

5.     yield 1 # 第一次进入, 先输出 first ，然后返回 1

7.     print("second")

8.     yield 10 # 第二次进入, 先输出 second ,然后返回 10

10.     print("three") # 第三次进入, 先输出 three , 但是没有找到 yield 返回的句子, 所以会报错

12. t = test()

13. v = t.__next__() # 第一次进入, 输出 first , v = 1

14. v = t.__next__() # 第二次进入, 输出 second , v = 10

15. v = t.__next__() # 第三次进入, 输出 three , 然后报错

例子 2 :

1. # 使用 .send() 向生成器发送信息, .send(None) 等价于 .__next__()

2. def test():

4.     print("a")

5.     m = yield 1 # 第一次进入，先输出 a , 然后返回 1, 这里 m 接收下一次进入时 send() 传入的值

7.     print(m)

8.     print("b")

10.     n = yield 2 # 第二次进入, 先执行 print(m) ,然后执行 print("b"), 最后返回 2

11.     print(n)

12.     print("c")

14.     # 第三次进入,先执行 print(n) ,然后执行 print("c"), 由于找不到 yield 返回的句子, 最后会报错

16. c = test()

17. v = c.send(None) # v = 1

18. v = c.send("hello") # v = 2

19. v = c.send("world") # 没有返回值，报错

1. 例子 3 ：

1. # 在 yield 中含有 return 关键词时，其作用是中止生成器

2. def test():

3.     print("hello")

4.     m = yield 1

5.     print(m)

6.     n = yield 2

7.     print(n)

8.     return

9.     print("you")

10.     yield 3

12. c = test()

13. v = c.send(None) # 第一次进入, 输出 hello, v = 1

14. print(v)

15. v = c.send("world") # 第二次进入, 输出 world, v = 2

16. print(v)

17. v = c.send("welcome") # 第三次进入, 输出 welcome , 遇到 return ，则直接被终止，报错提示为：··· StopIteration 

18. print(v)

例子 4 :

1. # 一般都不会手动调用 .__next__() 方法来访问,这样就可以避免在不知道 yield 有多少次返回时，最后出现如上报错的问题

2. def count(n):

3.     while n > 0:

4.         # 迭代器返回 n

5.         yield n

6.         # 下一次迭代器开始的位置

7.         n = n - 1

9. c = count(5)

10. for i in c:

11.     print(i)

相信通过以上 4 个例子就能够从总体上去知道了 yield 生成器的用法了，接来下通过 yield 来实现斐波那契数列。

1. def fibonacci(): 

2.     a = b = 1

3.     yield a # 第一次进入, 返回 1

4.     yield b # 第二次进入, 返回 1

5.     while True: 

6.         a,b = b,a + b 

7.         yield b # 第 i 次进入, 返回 a + b 的值

9. f = fibonacci()

10. c = 1

11. for i in f:

12.     print('第',c,'个数的值是：',i)

13.     c += 1

14.     # 输出前 10 个数的值

15.     if c > 10:

16.         break

最后在给出一个根据空格切割字符串的生成器，完整代码如下：

1. def split_by_split_blank_space(s):

2.     if s is None:

3.         return None

4.     if not isinstance(s, str):

5.         return None

6.     slen = len(s)

7.     if slen == 0:

8.         return None

9.     start = 0

10.     index = 0

11.     while index < slen:

12.         if s[index] == " ":

13.             if start < index:

14.                 yield s[start:index]

15.             index = index+1

16.             start = index

17.             continue

18.         index = index + 1

19.     if start < slen:

20.         yield s[start:]

23. result = split_by_split_blank_space("a b c d")

24. for i in result:

25.     print(i) # 依次输出 a b c d

yield 的出现主要是为了解决系统资源,避免一些不必要的内存占用。

我要打赏