第48讲：魔方方法——迭代器

一 迭代器的相关概念

• 迭代：每一次重复的过程被称作一次迭代，每一次迭代得到的结果，将会作为下一次迭代的初始值。
• 迭代器：提供迭代方法的容器。我们通常接触的迭代器有列表、元组、字符串和字典，他们都支持迭代操作。
• 举例：

• >>> for i in "FishC":
  ...     print(i)
  ...
  F
  i
  s
  h
  C

  例子中字符串就是一个容器，同时也是一个迭代器；for语句的作用就是触发这个迭代器的迭代功能，每次从容器里面一次拿出一个数据就是迭代操作。

• 内置函数：
  • iter()
    • 功能：用来生成迭代器。
    • 语法：iter(object[, sentinel])
    • 参数：
      • bject -- 支持迭代的集合对象。
      • sentinel -- 如果传递了第二个参数，则参数 object 必须是一个可调用的对象（如，函数），此时，iter 创建了一个迭代器对象，每次调用这个迭代器对象的__next__()方法时，都会调用 object。
    • 返回值：迭代器对象。
  • next()
    • 功能：next() 返回迭代器的下一个项目，下一个项目不存在会抛出StopIteration的异常。next() 函数要和生成迭代器的iter() 函数一起使用。

    • 语法：next(iterable[, default])

    •  

      参数说明：
      • iterable -- 可迭代对象
      • default -- 可选，用于设置在没有下一个元素时返回该默认值，如果不设置，又没有下一个元素则会触发 StopIteration 异常。
    • 返回值：返回对象帮助信息。
  • 举例：

  • >>> next(it)
    Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
    StopIteration
    >>> string = "FishC"
    >>> it = iter(string)
    >>> next(it)
    'F'
    >>> while True:
    ...     try:
    ...             each = next(it)
    ...     except StopIteration:
    ...             break
    ...     print(each)
    ...
    i
    s
    h
    C
    >>>

• 对应的魔法方法：
  • __iter__()
  • __next__()
  • 举例：

  • >>> class Fibs:
    ...     def __init__(self):
    ...             self.a = 0
    ...             self.b = 1
    ...     def __iter__(self):
    ...             return self
    ...     def __next__(self):
    ...             self.a,self.b = self.b,self.a + self.b
    ...             return self.a
    ...
    >>> fibs = Fibs()
    >>> for each in fibs:
    ...     if each < 20:
    ...             print(each)
    ...     else:
    ...             break
    ...
    1
    1
    2
    3
    5
    8
    13

  • 可以手动控制输出的斐波那契数列范围的代码：

  • >>> class Fibs:
    ...     def __init__(self,n = 10):
    ...             self.a = 0
    ...             self.b = 1
    ...             self.n = n
    ...     def __iter__(self):
    ...             return self
    ...     def __next__(self):
    ...             self.a,self.b = self.b,self.a+self.b
    ...             if self.a > self.n:
    ...                     raise StopIteration
    ...             return self.a
    ...
    >>> fibs = Fibs()
    >>> for each in fibs:
    ...     print(each)
    ...
    1
    1
    2
    3
    5
    8
    >>> fibs = Fibs(100)
    >>> for each in fibs:
    ...     print(each)
    ...
    1
    1
    2
    3
    5
    8
    13
    21
    34
    55
    89
    >>>