一、介绍
在Python中,迭代器是一种访问集合元素的方式,可以用于遍历数据集中的元素,而不需要事先知道集合的大小。迭代器可以被用于循环语句中,例如for循环,来遍历集合中的每个元素。
Python中的迭代器是一个实现了迭代器协议(Iterator Protocol)的对象。迭代器协议包含了两个方法:
__iter__()
:返回迭代器对象本身。__next__()
:next()方法返回容器中的下一个元素。当容器中的所有元素都被访问完毕后,next()方法会引发StopIteration异常,从而结束迭代。
迭代器的优点是可以按需生成元素,避免一次性加载所有元素造成的内存浪费;同时,迭代器可以实现惰性求值,避免不必要的计算,提高程序效率。
二、实例
# -*- coding: UTF-8 -*-
class MyIterator:
def __init__(self, data):
self.data = data
self.index = -1
def __iter__(self):
return self
def __next__(self):
self.index += 1
if self.index < len(self.data):
return self.data[self.index]
raise StopIteration
my_list = [1, 2, 3, 4, 5]
iterator = MyIterator(my_list)
for item in iterator:
print(item)