python迭代器

可迭代对象

list,str,tuple etc. —>for…in…遍历—>遍历(迭代)

li = [1, 2, 3]
for i in li:
    print(i, end=' ')

迭代器协议

对象必须提供一个next方法，执行该方法要么返回迭代中的下一项，要么引起StopIteration异常，以终止迭代(只能往下走，不可以回退)
实现了迭代器协议的对象就是可迭代对象

• 通过在对象内部定义一个__iter__方法

li = [1, 2, 3]
li_iter = li.__iter__()
print(li_iter.__next__())
'''
1
'''
print(li_iter.__next__())
'''
2
'''
print(li_iter.__next__())
'''
3
'''
print(li_iter.__next__())
'''
StopIteration  
(报错
'''

• 也可以用next():

li = [1, 2, 3]
li_iter = li.__iter__()
print(next(li_iter))
'''1'''
print(next(li_iter))
'''2'''
print(next(li_iter))
'''3'''

可迭代对象测试

使用instance()来判断一个对象是否可迭代
collections

from collections import Iterable

print(isinstance([], Iterable))
print(isinstance(str(), Iterable))
print(isinstance({}, Iterable))
print(isinstance(set(), Iterable))
print(isinstance(123, Iterable))
print(isinstance(True, Iterable))
'''
True
True
True
True
False
False
'''

自定义一个类，测试该类的可迭代性

from collections import Iterable

class MyClass:
    def __init__(self):
        self.names = []

    def add(self, name):
        self.names.append(name)

my_class = MyClass()
my_class.add("Tom")
my_class.add("Jack")
my_class.add("Lucy")
print("是否为可迭代对象：", isinstance(my_class, Iterable))
'''
是否为可迭代对象： False
'''

回顾之前说的__iter__方法，其可以为我们提供一个迭代器
在迭代一个可迭代对象的时候，实际上就是获取该对象提供的一个迭代器，然后通过该迭代器以此获取该对象的每一个数据

from collections import Iterable

class MyClass:
    def __init__(self):
        self.names = []

    def add(self, name):
        self.names.append(name)

    def __iter__(self):
        return None


my_class = MyClass()
my_class.add("Tom")
my_class.add("Jack")
my_class.add("Lucy")
print("是否为可迭代对象：", isinstance(my_class, Iterable))
for tmp in my_class:
    print(tmp, end=' ')
'''
是否为可迭代对象： True

TypeError: iter() returned non-iterator of type 'NoneType'
'''

for…in…循环的本质

就是通过iter()函数获取可迭代对象的Iterable的迭代器，然后对获取到的迭代器不断调用next()方法来获取下一个值并将其赋值，当遇到StopIteration的异常后，退出。

应用场景

迭代器的核心就是通过next()函数调用返回下一个数据值。如果每次返回的数据值不是在一个已有的数据集合中读取的，而是通过程序按照一定规律计算生成。那么也就意味着可以不用依赖一个已有的数据集合，namely，无需将所有的迭代对象数据一次性缓存下来供后续使用。这样可以节省大量的存储(内存)空间。
demo：斐波那契数列
通过for…in…的方式来遍历斐波那契数列中的前n个数
通过迭代来实现，每次迭代都可以通过数学计算来生成下一个数

class FibIterable(object):
    '''斐波那契数列迭代器'''

    def __init__(self, n):
        # 记录生成的斐波那契数列的个数
        self.n = n
        # 记录当前记录的索引
        self.current_index = 0
        # 记录两个初始值
        self.num1 = 0
        self.num2 = 1

    def __next__(self):
        '''调用next()函数来获取下一个数'''
        if self.current_index < self.n:
            num = self.num1
            self.num1, self.num2 = self.num2, self.num1 + self.num2
            self.current_index += 1
            return num
        else:
            raise StopIteration

    def __iter__(self):
        return self


fib = FibIterable(10)
for num in fib:
    print(num, end=' ')
'''
0 1 1 2 3 5 8 13 21 34 
'''