Python 迭代器
Python迭代器
迭代器是一个包含可数个值的对象。
迭代器是一个可以迭代的对象,这意味着您可以遍历所有值。
从技术上讲,在Python中,迭代器是一个实现迭代器协议的对象,它由方法__iter__()
和__next__()
。
Iterator 和 Iterable
列表,元组,字典和集都是可迭代的对象。它们是可迭代的容器,可以从中获取迭代器。
所有这些对象都有一个获取获取迭代器iter()
方法:
实例
从元组返回一个迭代器,并打印每个值:
mytuple = ("apple", "banana", "cherry") myit = iter(mytuple) print(next(myit)) print(next(myit)) print(next(myit))
字符串甚至都是可迭代的对象,并且可以返回迭代器:
实例
PYTHON标题
mystr = "banana" myit = iter(mystr) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit))
循环遍历
我们还可以使用for
循环遍历可迭代对象:
for
循环实际创建了迭代器对象,并且在每个循环中执行next()
方法。
创建一个迭代器
要创建一个对象或类作为迭代器,您必须实现方法 __iter__()
和 __next__()
对象。
正如您在Python 面向对象一章中所了解到的,所有类都有一个名为的函数 __init__()
,它允许您在创建对象时进行一些初始化。
__iter__()
方法行为类似,您可以执行操作(初始化等),但必须始终返回迭代器对象本身。
__next__()
方法还允许您执行操作,并且必须返回序列中的下一个项目。
实例
创建一个返回数字的迭代器,从1开始,每个序列将增加1(返回1,2,3,4,5等):
class MyNumbers: def __iter__(self): self.a = 1 return self def __next__(self): x = self.a self.a += 1 return x myclass = MyNumbers() myiter = iter(myclass) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter))
StopIteration关键字
如果您有足够的next()
语句,或者如果它在for循环中使用上面的示例将永远继续 。
为了防止迭代继续进行,我们可以使用该 StopIteration
语句。
在__next__()
方法中,如果迭代完成指定的次数,我们可以添加一个终止条件来引发错误:
实例
在迭代20次后停止:
class MyNumbers: def __iter__(self): self.a = 1 return self def __next__(self): if self.a <= 20: x = self.a self.a += 1 return x else: raise StopIteration myclass = MyNumbers() myiter = iter(myclass) for x in myiter: print(x)