python小demo——用自制的迭代器实现斐波那契数列

如果一个类中重写了__iter__和__next__这两个方法，那么这个类就是一个迭代器类
使用for语句遍历的原理就是利用了迭代器，所以可以使用for语句来访问自制的迭代器
代码：

class Fib:
	def __init__(self):
		self.a = 0
		self.b = 1
	def __iter__(self):
		return self# 本身就是迭代器，所以返回本身
	def __next__(self):
		temp = self.b
		self.b = self.a + self.b
		self.a = temp
		return self.a

for的每一次循环实际上就是在调用了__next__这个方法，得到下一个斐波那契数
for遍历：

f = Fib()
for each in f:
	if (each > 100):
		break
	else:
		print(each)

这里要注意加个if判断，不然就一直迭代下去了
或者也可以将Fib类改成这样：

class Fib:
	def __init__(self, n = 15):
		self.a = 0
		self.b = 1
		self.n = n
		self.count = 0
	def __iter__(self):
		return self# 本身就是迭代器，所以返回本身
	def __next__(self):
		temp = self.b
		self.b = self.a + self.b
		self.a = temp
		self.count += 1
		if self.count > self.n:
			raise StopIteration# 超过范围就抛异常，停止迭代
		else:
			return self.a

这样就不用再for那里处理了