学习笔记---python中对象的迭代 3--迭代器

迭代器

之前让对象也可以迭代，但是用到了两个类对象ClassA，ClassB。
ClassB中的__next__方法把ClassA中的列表拿过来，在返回给ClassA再给for。
直接将__iter__和__next__方法放在同一个对象中就可以了， __iter__的返回值就是对象自己，__next__的返回值就是需要遍历的属性
这就叫迭代器

class ClassA(object):

    def __init__(self):
        self.contains = list( )
        self.count = 0

    def addList(self, item):
        self.contains.append(item)
#如果想要一个对象成为可以迭代的对象，即可以使用for，必须使用__iter__方法
    def __iter__(self):
        return self

    def __next__(self):
        if self.count < len(self.contains):
            res = self.contains[self.count]
            self.count += 1
            return  res
        else :
            raise StopIteration #抛出异常


classIterable = ClassA( )

classIterable.addList('物品1')
classIterable.addList('物品2')
classIterable.addList('物品3')

for temp in classIterable:
    print(temp)

一个对象可以迭代，一定是迭代器
一个对象是迭代器，一定可以迭代
可以迭代的不一定是迭代器

迭代器的作用
迭代器存储的是生成数据的方法，可以占用极少的内存空间