可迭代的对象
使用 iter 内置函数可以获取迭代器的对象。 如果对象实现了能返回迭代器的 __iter__ 方法, 那么对象就是可迭代的。序列都可以迭代; 实现了 __getitem__ 方法, 而且其参数是从零开始的索引, 这种对象也可以迭代。
迭代器
迭代器要实现 __next__ 方法,返回单个元素,如果没有元素了, 那么抛出 StopIteration 异常。 此外还要实现__iter__ 方法, 返回迭代器本身。因此迭代器也可以迭代。
class A:
def __init__(self, data):
self.data = data
def __iter__(self):
return AIterator(self.data)
def __repr__(self):
return 'A' + str(self.data)
class AIterator:
def __init__(self, data):
self.data = data
self.index = 0
def __next__(self):
try:
d = self.data[self.index]
except IndexError:
raise StopInteration()
self.index += 1
return d
def __iter__(self):
return self
>>> a = A([1,2,3,4,5,6])
>>> it = iter(a)
>>> next(it)
1
>>> next(it)
2
>>> next(it)
3
>>> a
A[1, 2, 3, 4, 5, 6]
迭代器可以迭代, 但是可迭代的对象不是迭代器。
构建可迭代的对象和迭代器时经常会出现错误, 原因是混淆了二者。
可迭代的对象:有个 __iter__ 方法, 每次都实例化一个新的迭代器。可迭代的对象必须实现 __iter__ 方法, 但不能实现 __next__。为了“支持多种遍历”, 必须能从同一个可迭代的实例中获取多个独立的迭代器, 而且各个迭代器要能维护自身的内部状态。
迭代器:要实现 __next__ 方法, 返回单个元素, 此外还要实现__iter__ 方法, 返回迭代器本身