欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python3高级 一 迭代器

程序员文章站 2024-03-21 22:52:58
...

迭代器

迭代是访问元素集的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退

可迭代对象

我们已经知道可以对list、tuple、str等类型的数据使用for…in…的循环语法从其中依次拿到数据进行使用,我们把这样的过程称为遍历,也叫迭代

但是,是否所有的数据类型都可以放到for…in…的语句中,然后让for…in…每次从中取出一条数据供我们使用,即供我们迭代吗?

  • 可以使用 isinstance()判断一个对象是否是 Iterable 对象
a = list()   	# Iterable可迭代对象
print(isinstance(a, Iterable))  # isinstance(要判断的对象,类型)

输出:True

b = 100
print(isinstance(b, Iterable))   #  输出 False
# int整型不是iterable,即int整型不是可以迭代的

可迭代对象的本质

我们分析对可迭代对象进行迭代使用的过程,发现每迭代一次(即在for…in…中每循环一次)都会返回对象中的下一条数据,一直向后读取数据直到迭代了所有数据后结束。那么,在这个过程中就应该有一个“人”去记录每次访问到了第几条数据,以便每次迭代都可以返回下一条数据。我们把这个能帮助我们进行数据迭代的“人”称为迭代器(Iterator)

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

可迭代对象通过__iter__方法向我们提供一个迭代器,我们在迭代一个可迭代对象的时候,实际上就是先获取该对象提供的一个迭代器,然后通过这个迭代器来依次获取对象中的每一个数据.

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

from collections import Iterable

class Classmate():
    def __init__(self):
        self.names = list()

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

    def __iter__(self):  # 该方法的返回值是一个迭代器
        print("我是iter方法")
       # return self

iter()函数与next()函数

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

a = [11, 22, 33, 44, 55]
a_iter = iter(a)  # 获得列表a的迭代器
print(next(a_iter))  # next(迭代器) 取出迭代器中的数据
print(next(a_iter))

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

迭代器Iterator

我们已经知道,迭代器是用来帮助我们记录每次迭代访问到的位置,当我们对迭代器使用next()函数的时候,迭代器会向我们返回它所记录位置的下一个位置的数据。实际上,在使用next()函数的时候,调用的就是迭代器对象的__next__方法.所以,我们要想构造一个迭代器,就要实现它的__next__方法。但这还不够,python要求迭代器本身也是可迭代的,所以我们还要为迭代器实现__iter__方法,而__iter__方法要返回一个迭代器,迭代器自身正是一个迭代器,所以迭代器的__iter__方法返回自身即可。

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

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

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

    def __iter__(self):     # 有__iter__方法,该类的实例对象是一个可迭代对象
        myiterator = MyIterator(self)
        return myiterator


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

    def __next__(self):    #  有__next__方法 ,有__iter__方法,所以该类的实例对象是一个迭代器
        if self.current < len(self.mylist.items):
            item = self.mylist.items[self.current]
            self.current += 1
            return item
        else:
            raise StopIteration

    def __iter__(self):  # 为什么该类要有__iter__方法,因为迭代器自身也要求是可迭代的
        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)

for…in…循环的本质

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

迭代器的应用场景

我们发现迭代器最核心的功能就是可以通过next()函数的调用来返回下一个数据值。如果每次返回的数据值不是在一个已有的数据集合中读取的,而是通过程序按照一定的规律计算生成的,那么也就意味着可以不用再依赖一个已有的数据集合,也就是说不用再将所有要迭代的数据都一次性缓存下来供后续依次读取,这样可以节省大量的存储(内存)空间。

from collections import Iterable


class Classmate():
    def __init__(self):
        self.names = list()

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

    def __iter__(self):  # 该方法的返回值是一个迭代器
        print("我是iter方法")
        return ClassIterble(self)


class ClassIterble():
    def __init__(self, item):
        self.item = item
        self.c = 0

    def __iter__(self):
        pass

    def __next__(self):
        if self.c < len(self.item.names):
            re = self.item.names[self.c]
            self.c += 1
            return re
        else:
            raise StopIteration


c = Classmate()
c.add("1")
c.add("2")
c.add("3")
for name in c:
    print(name)

改进一下,让迭代器返回自身,一个类自己是可迭代对象,也是迭代器

from collections import Iterable
from collections import Iterator
import time

class Classmate:
    def __init__(self):
        self.names = list()
        self.count = 0


    def add(self, item):
        self.names.append(item)

    def __iter__(self):
        return self

    def __next__(self):
        if self.count < len(self.names):
            re = self.names[self.count]
            self.count += 1
            return re
        else:
            raise StopIteration


c = Classmate()

c.add("第一")
c.add("第二")
c.add("第三")
c.add("第四")
for i in c:
    print(i)
    time.sleep(1)
lsi = [i for i in c]
print(lsi)