迭代器详解

Python三大器之一

                                                                                            迭代器详解

---

1.问题由来：

            类似于遍历一个数组，字符串 ，元组，字典一样，可以遍历一个对象中的属性存储的列表吗？

            回答：不能

for  e  in range(5):
    print(e)  #  打印出来 1 2 3 4 5                                          【成功执行】

class ClassMate(object):
    def __init__(self):
        self.studentName = list()

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

mate = ClassMate()
mate.add('张三')
mate.add('李四')
mate.add('王五')
for name in mate:
    print(name)                                          【报错，执行抛出异常】                                

以上遍历类的操作报错，抛出异常  TypeError: 'ClassMate' object is not iterable。

2.解决办法

        （1）理解异常 ->[TypeError: 'ClassMate' object is not iterable],这提示的是该对象不是可迭代对象。

            在Python中，一个对象中包含 __iter__魔法方法，就可以定为可迭代对象。

        （2）添加__iter__魔法方法到类中

            可以看到依然出现了问题，提示的是迭代器返回了不是迭代器类型的NoneType类型的对象，

简而言之就__iter__要返回一个迭代器类型的对象

            （3）接着返回一个迭代器类型的对象（迭代器必须包含__iter__ 和 __next__两个魔法方法）

                上图的黄色框中就是添加的返回迭代类型的对象和，迭代器类。当代码执行时，就可以看到每秒都会打印出返回值

看到这里，我们大概就明白了。迭代器的执行就是每次返回__next__返回的结果。那我们对iterOp类中的__next__方法改造一下，

这里基本可以遍历对象中指定的内容了。但是遍历结束后提交的是None，这里特别说明，对于迭代而言，当捕获到StopIteration异常，迭代就会停止

所以在if之后的else之后，抛出StopIteration异常   ，就可以结束迭代。至此迭代器完成。

import time
class ClassMate(object):
    def __init__(self):
        self.studentName = list()
    def add(self,name):
        self.studentName.append(name)
    def __iter__(self):
        return iterOp(self)
class iterOp(object):
    def __init__(self,obj):
        self.obj = obj
        self.count = 0
    def __iter__(self):
        pass
    def __next__(self):
        # self.obj.studentName[0]
        time.sleep(1)
        if self.count < len(self.obj.studentName):
            res = self.obj.studentName[self.count]
            self.count += 1
            return res
        else:
            raise StopIteration

mate = ClassMate()
mate.add('张三')
mate.add('李四')
mate.add('王五')
for name in mate:    
    print(name)

2.代码简化

通过上面迭代器必要组件的分析，包含有__iter__,__next__，方法的类都可以看做迭代器。所以讲可以将迭代器类简化到目标类中。

import time
class ClassMate(object):
    def __init__(self):
        self.count = 0    #  计数器初始化
        self.studentName = list()
    def add(self,name):
        self.studentName.append(name)
    def __iter__(self):
        return self                # 返回迭代器对象，self就是对象的引用。（自身此时包含__iter__和__next__方法，自己也
                                   #   就是迭代器）
    def __next__(self):
        # self.obj.studentName[0]
        time.sleep(1)
        if self.count < len(self.studentName):
            res = self.studentName[self.count]
            self.count += 1
            return res
        else:
            raise StopIteration
mate = ClassMate()
mate.add('张三')
mate.add('李四')
mate.add('王五')
for name in mate:    
    print(name)

3.迭代器总结

• for 遍历的语法执行分析