Python基础学习——迭代器与生成器

一、迭代器

迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。迭代器有两个基本的方法：iter() 和 next()。字符串，列表或元组对象都可用于创建迭代器。

list=[4,5,6,7]
it=iter(list) #创建迭代器对象
print(next(it))
'''
上一条输出4，下一个循环输出5，6，7
因为在next（it）后，迭代器it移动到了5
'''
for x in it:
  print(x)

创建一个迭代器

__iter__() 方法返回一个特殊的迭代器对象， 这个迭代器对象实现了 __next__()方法并通过 StopIteration 异常标识迭代的完成。

通过在类中添加 __iter__函数，向系统说明这是一个可迭代对象。通过在类中添加 __next__函数，向系统提供该可迭代对象的迭代算法

class fab(): #斐波那契数列
  n=0
  def __init__(self,a):
    self.max=a
    self.n=0
    self.a=0 
    self.b=1 
  def __iter__(self):
    return self
  def __next__(self):
    if self.n<self.max:
      self.n+=1
      center=self.b
      self.b=self.a+self.b
      self.a=center
      return self.a
    else:
      raise StopIteration
for i in fab(5): #输出斐波那契数列的前五位
  print(i)

StopIteration 异常用于标识迭代的完成，防止出现无限循环的情况，在 __next__()方法中我们可以设置在完成指定循环次数后触发 StopIteration 异常来结束迭代。

二、生成器

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

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

def fab(n): #生成器函数
  a,b=0,1
  out=1
  while True:
    if out>n:
      return 
    yield a
    a,b=b,b+a
    out+=1
f=fab(5) #返回迭代器，由生成器返回
print(f) #输出<generator object fab at 0x046AFA00>
for i in range(0,5):
  print(next(f))