迭代器

2019-03-26

迭代器（高级）

首先介绍一个用迭代器实现for功能的例子

# 此示例示意用迭代器遍历一个列表
L = [2, 3, 5, 7]
# 1. 先让L提供一个迭代器
it = iter(L)
# 循环从迭代器中获取数据,直到接收到StopIteration异常为止
while True:
    try:
        x = next(it)
        print(x)
    except StopIteration:
	    break

# ------- 以上多条语句 可以用如下的for循环来实现
for x in L:
    print(x)

什么是迭代器:

可以通过next函数取值的对象就是迭代器
迭代器协议
迭代器协议是指对象能够使用next函数获取下一项数据,在没有下一项数据时触发一个StopIteration异常来终止迭代的约定
迭代器协议实现的方法:

方法名                             对应函数
def __next__(self):               next(obj)

什么是可迭代对象:

是指能用iter(obj) 函数返回迭代器的对象叫可迭代对象
可迭代对象内部需要定义iter(self)方法来返回迭代器

方法名                               对应函数
def __iter__(self):                 iter(obj)

例子

# 此示例用来说明迭代器和可迭代对象的差别

class MyList:
    def __init__(self, iterable=()):
        self.data = [x for x in iterable]
    def __repr__(self):
        return 'MyList(%s)' % self.data
    def __iter__(self):
        '''有此方法的对象可以用iter(obj) 来获取迭代器'''
        return MyListIterator(self.data)  # "迭代器"

class MyListIterator:
    '''此类用于创建一个迭代器,此迭代器可以用来访问
    MyList类型的对象'''
    def __init__(self, lst):
        self.data2 = lst
        self.index = 0  # 用来记录此迭代器当前访问的地点
    def __next__(self):
        if self.index >= len(self.data2):
            raise StopIteration
        r = self.data2[self.index]
        self.index += 1
        return r

myl = MyList([1, -2, 3, -4, 5])
# for x in myl:  # <<<可以吗??  #如果MyList对象无__iter__ 方法则不行
#     print(x)

it = iter(myl)  # it = myl.__iter__()
while True:
    try:
        x = next(it)  # x = it.__next__()
        print(x)
    except StopIteration:
        break