python3 迭代器

1. 想要了解 迭代器 ,需要先 知道什么是 可迭代对象。

  • 简单点说,可以直接作用于for循环的对象,称之为可迭代对象(Iterable)。

    python3 迭代器_第1张图片
    1.png

    可迭代对象,一定实现了__iter__方法。isinstance內建函数,可以判断一个对象是否是某个类的实例对象。通过运行代码我们可以看到,返回了True. 表示 it 这个对象是一个可迭代对象。

    __iter__ 这个方法只有返回一个特殊的对象才有实际的意义。(这个特殊对象,可以调用next方法,不断返回一个值,没有值的时候抛出StopIteration异常),这里所说的特殊对象,就是迭代器(Iterator)。
    iter函数的作用:调用 参数对象的 __iter__方法 返回一个 迭代器对象(Iterator)。

2. 迭代器(Iterator)

  • 简单点说,可以调用__next__方法,不断返回一个值的对象称之为迭代器.
  • 迭代器要求必须有__iter__方法和__next__方法。换句话说,也就是,如果你想自己定义一个迭代器,那么你必须定义__iter__方法和__next__方法。

自定义 迭代器:

from collections import Iterable, Iterator

class test(object):
    def __init__(self, b):
        self.a = 0
        self.b = b

    def __iter__(self):
        return self

    def __next__(self):
        if self.a < self.b:
            i = self.a
            self.a += 1
            return i
        else:
            raise StopIteration()

it = test(10)
print(isinstance(it, Iterable))
print(isinstance(it, Iterator))

# next(it)  # 实际上 调用 it 对象的 __next__ 方法。

for temp1 in it:
    print(temp1)

for temp2 in it:
    print(temp2)

可见:迭代器 一定是 可迭代对象。列表、元组、字符串、字典、集合 这些虽然是 可迭代对象,但是不是 迭代器。iter() 函数可以很方便的把他们转换为迭代器。

for循环的工作原理:调用iter()函数,获取一个迭代器对象,然后迭代器调用__next__()方法不断返回一个值,直到遇到StopIteration这个异常 ,停止循环。

注意:迭代器 在 返回所有值之后,就空了,没有了。it这个对象,就好比是一个盛满水的水壶,调用__next__方法相当于 倒水,水倒完了,水壶自然就空了

迭代器是一次性的,只能作用于一次for循环,当然 这不是绝对的,可以 自己修改成 无限次。如何修改呢?
只需要在 __iter__ 方法中 加上 self.a = 0 这行代码即可。

你可能感兴趣的:(python3 迭代器)