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()