python : 自定义可迭代类，iter ,next的作用

逃离我推掉我的手 2023-01-09 12:56 104阅读 0赞

转自：[https://blog.csdn.net/Strive\_For\_Future/article/details/91306814?utm\_medium=distribute.pc\_relevant\_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control&depth\_1-utm\_source=distribute.pc\_relevant\_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control][https_blog.csdn.net_Strive_For_Future_article_details_91306814_utm_medium_distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control_depth_1-utm_source_distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control]

python\_迭代器协议

一、定义

1、迭代器协议是指：对象必须提供一个next方法，执行方法要么返回迭代器中的下一项，要么就引起一个StopIteration异常，以终止迭代（单向运行：只能前进，不能倒退）；

2、可迭代对象：实现了迭代器协议的对象（实现方式：对象内部定义了一个iter()方法）；

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

二、迭代器协议的实现

迭代器协议规定了对象必须提供一个next方法和iter方法。

正式的说法是：实现了\_\_iter\_\_方法的对象是可迭代对象，实现了\_\_next\_\_方法的对象是迭代器。

在python3中，如果定义了一个类，类中有 \_\_iter\_\_(self)，\_\_next\_\_(self)两个函数，则该类具有了迭代器的属性，

具体性质体现在：

如果该类的对象出现在一个for循环中，那么for循环会自动通过\_\_iter\_\_标志识别这是一个可迭代的类，继而自动调用\_\_next\_\_()函数

class Fab(object):
        def __init__(self,max,):
            self.max=max
            self.n,self.a,self.b=0,0,1
        def __iter__(self):
            return self
        def __next__(self):
            if self.n < self.max:
                out = self.b
                self.a,self.b=self.b,self.a+self.b
                self.n = self.n+1
                return out
            raise StopIteration()
        
    for i in Fab(6):
        print(i)

返回结果：

1
    1
    2
    3
    5
    8

总结：

1）通过在类中添加 \_\_iter\_\_函数，向系统说明这是一个可迭代对象。

2）通过在类中添加 \_\_next\_\_函数，向系统提供该可迭代对象的迭代算法

3）在代码执行过程中，for循环函数会自动检查系统信息，识别\_\_iter\_\_函数，然后自动调用对应的\_\_next\_\_函数，生成一个迭代器。

4）所以在定义一个可迭代类时，一般\_\_iter\_\_ 函数要与 \_\_next\_\_函数成对出现。\_\_iter\_\_函数向系统声明这个类可迭代，\_\_next\_\_定义了具体的迭代器。

5）\_\_iter\_\_ 与 \_\_next\_\_两个函数名不可改变，否则系统会不识别。

6）\_\_next\_\_函数的 return 在 if 判别命令的内部，每次执行\_\_next\_\_函数时，单次判别后直接输出。不满足判别条件时输出迭代终止。

[https_blog.csdn.net_Strive_For_Future_article_details_91306814_utm_medium_distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control_depth_1-utm_source_distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control]: https://blog.csdn.net/Strive_For_Future/article/details/91306814?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.control