1. 想要了解 迭代器 ,需要先 知道什么是 可迭代对象。
简单点说,可以直接作用于for循环的对象,称之为可迭代对象(Iterable)。
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 这行代码即可。