Python实现自定义可迭代对象

class Square():
    def __init__(self, start, stop):

        self.value = start - 1
        self.stop = stop


    def __iter__(self):

        return self


    def __next__(self):

        if self.value == self.stop:
            raise StopIteration
        self.value += 1
        return self.value ** 2


print('---' * 100)

for i in Square(1, 10):
    print(i)


#结果：
1
4
9
16
25
36
49
64
81
100

1.可迭代对象的本质就是可以向我们提供一个这样的中间“人”即迭代器帮助我们对其进行迭代遍历使用。

可迭代对象通过__iter__方法向我们提供一个迭代器，我们在迭代一个可迭代对象的时候，实际上就是

先获取该对象提供的一个迭代器，然后通过这个迭代器来依次获取对象中的每一个数据。

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

class MyList(object):
     def __init__(self):
             self.container = []
     def add(self, item):
             self.container.append(item)
     def __iter__(self):
             """返回一个迭代器"""
             # 我们暂时忽略如何构造一个迭代器对象
             pass

mylist = MyList()
from collections import Iterable
isinstance(mylist, Iterable)
True

# 这回测试发现添加了__iter__方法的mylist对象已经是一个可迭代对象了

2.iter()函数与next()函数

ist、tuple等都是可迭代对象，我们可以通过iter()函数获取这些可迭代对象的迭代器。然后我们可以对获取到的迭代器不断使用next()函数来获取下一条数据。iter()函数实际上就是调用了可迭代对象的__iter__方法。

'''
学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：725638078
寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！
'''
>>> li = [11, 22, 33, 44, 55]
>>> li_iter = iter(li)
>>> next(li_iter)
11
>>> next(li_iter)
22
>>> next(li_iter)
33
>>> next(li_iter)
44
>>> next(li_iter)
55
>>> next(li_iter)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>>

注意
当我们已经迭代完最后一个数据之后，再次调用next()函数会抛出StopIteration的异常，来告诉我们所有数据都已迭代完成，不用再执行next()函数了。

3.一个实现了__iter__方法和__next__方法的对象，就是迭代器。

class MyList(object):
    """自定义的一个可迭代对象"""
    def __init__(self):
        self.items = []

    def add(self, val):
        self.items.append(val)

    def __iter__(self):
        myiterator = MyIterator(self)
        return myiterator


class MyIterator(object):
    """自定义的供上面可迭代对象使用的一个迭代器"""
    def __init__(self, mylist):
        self.mylist = mylist
        # current用来记录当前访问到的位置
        self.current = 0

    def __next__(self):
        if self.current < len(self.mylist.items):
            item = self.mylist.items[self.current]
            self.current += 1
            return item
        else:
            raise StopIteration

    def __iter__(self):
        return self


if __name__ == '__main__':
    mylist = MyList()
    mylist.add(1)
    mylist.add(2)
    mylist.add(3)
    mylist.add(4)
    mylist.add(5)
    for num in mylist:
        print(num)

4.for…in…循环的本质

for item in Iterable 循环的本质就是先通过iter()函数获取可迭代对象Iterable的迭代器，然后对获取到的迭代器不断调用next()方法来获取下一个值并将其赋值给item，当遇到StopIteration的异常后循环结束。

结尾给大家推荐一个非常好的学习教程，希望对你学习Python有帮助！

Python基础入门教程推荐：更多Python视频教程-关注B站：Python学习者
https://www.bilibili.com/video/BV1LL4y1h7ny?share_source=copy_web

Python爬虫案例教程推荐：更多Python视频教程-关注B站：Python学习者
https://www.bilibili.com/video/BV1QZ4y1N7YA?share_source=copy_web