python中的迭代器

目录

一.什么是迭代器

二、for循环原理

三、迭代器作用

四、应用场景

---

一.什么是迭代器

拥有__iter__方法和__next__方法的对象就是迭代器

 

1、迭代

迭代是访问集合元素的一种方式，可以将某个数据集内的数据“一个挨着一个的取出来”，就叫做迭代。

 

2、可迭代协议

协议就是互相规定好的。可迭代协议的定义非常简单，就是内部实现了__iter__方法。

 

3、迭代器协议：必须拥有__iter__方法和__next__方法。

 

• 列举能被for 循环的内容

list、dic、str、set、tuple、f.open()、range()、enumerate

 

• dir()方法获取对应数据类型的所有方法。

dir() 获取对应数据类型的所有方法，带有下划线的方法叫双下方法。

例如：print(dir(list))

 

• 查看 [],{},''，range() 共有的方法:使用集合的交集进行获取。

找到了__iter__ 方法。有此方法的对象就是一个可以被迭代的对象。

 

• __iter__方法的作用是返回一个迭代器。

 

二、for循环原理

1.for循环一个可迭代的对象(实现__iter__方法)

2.__iter__方法返回一个迭代器(迭代器实现了__iter__方法和__next__方法)

3.for先判断对象地方可迭代，然后调用迭代器的__next__方法获取值。

 

三、迭代器作用

节约内存，取得时候再生成数据，python2.7的时候range()方法就立刻生成了数据，占用了大量的内存空间。

四、应用场景

1、数据类型转换 例如 list和元组之间的转换底层就是使用的迭代器

 

import time


class Names(object):
    def __init__(self):
        self.lst = []

    def add_name(self, name):
        self.lst.append(name)

    def __iter__(self):
        return NamesIterator(self.lst)


class NamesIterator(object):
    def __init__(self, lst):
        self.lst = lst
        self.current_num = 0

    def __iter__(self):
        pass

    def __next__(self):
        if self.current_num < len(self.lst):
            name = self.lst[self.current_num]
            self.current_num += 1
            return name
        else:
            raise StopIteration


names = Names()
names.add_name('ls')
names.add_name('zs')
names.add_name('ww')

for i in names:
    time.sleep(1)
    print(i)

输出为：

ls
zs
ww

如上代码所示，用__iter__方法赋予了Names类迭代属性，并返回了NamesIterator类。其中NamesIterator类添加了__next__方法，使其变成一个完整的迭代器，当然我们也可以直接在Names类的添加__next__方法。所以最后names对象实例继承了Names类的可迭代属性，可以被for使用