python 迭代器

迭代器

迭代器协议是指：
1、对象必须提供一个next方法，执行该方法要么返回迭代的下一项，要么就引起一个Stoplteration异常，来终止迭代（只能往后走不能往前退）

2、可迭代对象: 实现了迭代器协议的对象（通过在对象内部定义一个__iter__（）方法）

3、协议是一种约定，可迭代对象实现了迭代器协议，python的内部工具使用迭代器协议访问对象，内部工具是指如for循环，sum，min，max函数等

python当中一切皆对象，python当中所有的数据类型只要遵循迭代器协议去生成的，那他就是可迭代对象

（字符串、列表、元组、字典、集合、文件对象）这些都不是可迭代对象，只是for循环调用了他们内部的__iter__方法，把它们变成了可迭代对象，跟索引没有关系

可迭代对象

可以对list、tuple、str等类型的数据使用for…in…的循环语法从其中依次拿到数据进行使用，我们把这样的过程称为遍历，也叫迭代。

li = [1,2,3]
li_iter = iter(li)    #获取可迭代对象的迭代器
print(li_iter)
while True:
    try:
        print(next(li_iter))
    except StopIteration:
        pass
# print(next(li_iter))   #再读一次会报错

<list_iterator object at 0x000002AD62314278>
1
2
3

​ 我们已经知道，可以直接作用于for循环的数据类型有以下几种：一类是集合数据类型，如list、tuple、dict、set、str等；一类是generator，包括生成器和带yield的generator function。这些可以直接作用于for循环的对象统称为可迭代对象：Iterable。

​ 可以使用isinstance()判断一个对象是否是Iterable对象：

>>> from collections import Iterable
>>> isinstance([], Iterable)
True
>>> isinstance({}, Iterable)
True
>>> isinstance('abc', Iterable)
True
>>> isinstance((x for x in range(10)), Iterable)
True
>>> isinstance(100, Iterable)
False

​ 可以被next()函数调用并不断返回下一个值的对象称为迭代器：Iterator。

​ 可以使用isinstance()判断一个对象是否是Iterator对象：

>>> from collections import Iterator
>>> isinstance((x for x in range(10)), Iterator)
True
>>> isinstance([], Iterator)
False
>>> isinstance({}, Iterator)
False
>>> isinstance('abc', Iterator)
False
>>> isinstance(iter([]), Iterator)   #iter() 等价于 ().__iter__
True
>>> isinstance(iter("abc"), Iterator)
True

​ 生成器都是Iterator对象，但list、dict、str虽然是Iterable，却不是Iterator。

可迭代对象的本质

​ 可迭代对象通过__iter__方法向我们提供一个迭代器，我们在迭代一个可迭代对象的时候，实际上就是先获取该对象提供的一个迭代器，然后通过这个迭代器来依次获取对象中的每一个数据.

​ 那么也就是说，一个具备了__iter__方法的对象，就是一个可迭代对象。

>>> class MyList(object):
...     def __init__(self):
...             self.container = []
...     def add(self, item):
...             self.container.append(item)
...     def __iter__(self):
...             """返回一个迭代器"""
...             # 我们暂时忽略如何构造一个迭代器对象
...             pass  
...
>>> mylist = MyList()
#可以用mylist.add一下
>>> from collections import Iterable
>>> isinstance(mylist, Iterable)
True
# 这回测试发现添加了__iter__方法的mylist对象已经是一个可迭代对象了

for…in…循环的本质

li = [1,2,3]
print(iter(li))
li_iter = iter(li)
# print(next(li_iter))
# print(next(li_iter))
# print(next(li_iter))
#print(next(li_iter))
for i in li_iter:
    print(i,end = " ")

<list_iterator object at 0x000001F03DC24278>
1 2 3

应用场景

​ 迭代器最核心的功能是可以通过next()函数的调用来返回下一个数据值。如果每次返回的数据值不是在一个已有的数据集合中读取的，而是通过程序按照一定的规律计算生成的，那么也就意味着可以不用再依赖一个已有的数据集合，也就是说不用再将所有要迭代的数据都一次性缓存下来供后续依次读取，这样可以节省大量的存储（内存）空间。

​ 举个例子，比如，数学中有个著名的斐波拉契数列（Fibonacci），数列中第一个数为0，第二个数为1，其后的每一个数都可由前两个数相加得到：0, 1, 1, 2, 3, 5, 8, 13, 21, 34, …

​ 现在我们想要通过for…in…循环来遍历迭代斐波那契数列中的前n个数。那么这个斐波那契数列我们就可以用迭代器来实现，每次迭代都通过数学计算来生成下一个数。

class FibIterator(object):
    """斐波那契数列迭代器"""
    def __init__(self, n):
        # n: int, 指明生成数列的前n个数
        self.n = n
        # current用来保存当前生成到数列中的第几个数了
        self.current = 0
        # num1用来保存前前一个数，初始值为数列中的第一个数0
        self.num1 = 0
        # num2用来保存前一个数，初始值为数列中的第二个数1
        self.num2 = 1

    def __next__(self):
        """被next()函数调用来获取下一个数"""
        if self.current < self.n:
            num = self.num1
            self.num1, self.num2 = self.num2, self.num1 + self.num2
            self.current += 1
            return num
        else:
            raise StopIteration   #抛出异常

    def __iter__(self):
        """迭代器的__iter__返回自身即可"""
        return self


if __name__ == '__main__':
    fib = FibIterator(10)
    for num in fib:
        print(num, end=" ")

0 1 1 2 3 5 8 13 21 34