14. 迭代器和生成器

迭代是Python最强大的功能之一,是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器有两个基本的方法:iter() 和 next()。

14.1 迭代器生成

字符串,列表或元组对象都可用于创建迭代器

list=[1,2,3,4]
it = iter(list)		# 创建迭代器对象
print(next(it))		# 输出迭代器的下一个元素
print(next(it))

14.2 迭代器遍历

list=[1,2,3,4]
it = iter(list)

(1) for循环

for i in it:
	print(i)

(2) while循环

import sys
while True:
	try:
		print(next(its))
	except StopIteration:
		sys.exit()

14.3 创建一个迭代器

把一个类作为一个迭代器使用需要在类中实现两个方法 iter() 与 next() 。
例如:

class MyNumbers:
	def __iter__(self):
		self.a = 1
		return self 
	def __next__(self):
		if self.a < 20:
			x = self.a
			self.a += 1
			return x
		else:
			raise StopIteration
	myclass = MyNumbers()
	myiter = iter(myclass) 
	print(next(myiter))
	for x in myiter:
				print(x)

14.4 生成器

在Python中,使用了yield的函数被称为生成器(generator)。跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。

在调用生成器运行的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息,返回yield的值, 并在下一次执行next()方法时从当前位置继续运行。

调用一个生成器函数,返回的是一个迭代器对象。

import sys
def fibonacci(n): 				# 生成器函数 - 斐波那契
a, b, counter = 0, 1, 0
while True:
	if (counter > n): 
               return
	yield a
	a, b = b, a + b
	counter += 1
f = fibonacci(10) 				# f是一个迭代器,由生成器返回生成
while True:
   try:
	print (next(f), end=" ")
except StopIteration:
	sys.exit()