迭代器
(1)定义
迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。迭代器不仅可以用于已知的数据结构,还可以自己构造迭代器。例如,ist 是可迭代对象,dict 也是可迭代对象。
当运行如下代码时:
example = [1,2,3,4,5]
for i in example:
print(i)
实际上,list是一个可迭代对象,其内部使用python内置函数iter()调用遍历对象的__iter__()获得一个迭代器,之后再循环对这个迭代器使用next()调用迭代器对象的__next__()。iter()只会被调用一次,而__next__()会被调用 n 次。
(2)优点
迭代器的优势有以下几点:
(1)降低内存消耗:迭代器可以逐个处理数据,不需要一次性将整个数据集加载到内存中,因此可以有效降低内存消耗。这点是源自于它的懒加载的特性,即每次只加载需要使用的内容。
(2)方便迭代:迭代器可以在不同的场景下使用,因此可以增加代码的灵活性。迭代器有两个基本的方法:iter() 和 next()。
例如对于list结构,可以使用next去访问。实际上
>>> list=[1,2,3,4]
>>> it = iter(list) # 创建迭代器对象
>>> print (next(it)) # 输出迭代器的下一个元素
1
>>> print (next(it))
2
或者自己定义一个迭代器:
class test(object):
def __init__(self, cnt):
self.val = 0
self.cnt = cnt
def __iter__(self):
return self
def __next__(self):
if self.cnt > 0:
self.cnt -= 1
self.val += 1
return self.val
else:
raise StopIteration
if __name__ == '__main__':
res = test(5)
for i in res:
print(i)
运行代码得到的结果是:1,2,3,4,5.因此迭代器不止运用在常见的数据结构中,也可以是自己定义的类。总而言之,需要是可迭代对象。