Python中的yield关键字分析：一个Generator

写作不易，资瓷一些呗！个人博客：https://raycoder.me

本文首发于Ray’s Blog

我们先来康一段代码：

>>> def count(n):
		for i in range(1, n+1):
			print(i)

>>> count(5)
1
2
3
4
5

这一小段代码执行了计数，输出了1-5。

不过，有时我们会有一些想♂FA，想要每次获取下一个数，如第一次执行时给你返回1，第二次返回2，第三次返回3……如此如此。

你可能会这么做：

>>> n = 0
>>> def addone():
		global n
		n+=1
		print(n)

>>> addone()
1
>>> n
1

但这样又暴露了一个缺点：{ {< udpoint “我们使用了一个全局变量” >}}。想要避免这种缺点又要实现计数的功能，yield就派上用场了。

想要实现Generator的功能，只需要对代码进行一处改动：把print()改成yield即可。

>>> n = 0
>>> def count(n):
		for i in range(1, n+1):
			yield i

>>> c = count(5)
>>> c
<generator object addone at 0x000002329F851048>
>>> 

你会发现，这个c的类型像是一个对象了。

>>> c = count(5)
>>> next(c)
1
>>> next(c)
2
>>> next(c)
3
>>> next(c)
4
>>> next(c)
5
>>> next(c)
Traceback (most recent call last):<