1、Python 迭代器(Iterator)
迭代器是一个包含数个值的对象。
迭代器是可以迭代的对象,这意味着您可以遍历所有值。
从技术上讲,在Python中,迭代器是实现迭代器协议的对象,该协议由方法__iter__()
和__next__()
组成。
例如:
迭代一个元组中元素:
mytuple = ("c", "python", "java") for x in mytuple: print(x)
2、迭代器(Iterator)和可迭代(Iterable)
列表,元组,字典和集合都是可迭代的对象。 它们是可迭代的容器,可以从中获得迭代器。
所有这些对象都有一个iter()
方法,该方法用于获取迭代器:
例如:
从元组返回一个迭代器,并输出每个值:
mytuple = ("c", "java", "python") myit = iter(mytuple) print(next(myit)) print(next(myit)) print(next(myit))
字符串也是可迭代对象,可以返回迭代器:
例如:
字符串可以当作字符列表:
mystr = "python" myit = iter(mystr) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit)) print(next(myit))
3、遍历迭代器
还可以使用for
循环来迭代可迭代对象:
例如:
迭代一个元组的值:
mytuple = ("c", "java", "python") for x in mytuple: print(x)
例如:
迭代字符串的字符:
mystr = "banana" for x in mystr: print(x)
for
循环实际上创建了一个迭代器对象,并为每个循环执行next()
方法。
4、定义一个迭代器
要创建一个对象/类作为迭代器,必须实现__iter__()
和__next__()
方法。
正如在Python 类和对象一章中所了解的那样,所有类都有一个名为__init__()
的函数,该函数可以在创建对象时进行一些初始化。
__iter__()
方法的行为类似,可以执行操作(初始化等),但必须始终返回迭代器对象本身。
__next__()
方法还允许你进行操作,并且必须返回序列中的下一项。
例如:
创建一个返回数字(从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))
5、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)
# 首先获得Iterator对象: it = iter([1, 2, 3, 4, 5]) # 循环: while True: try: # 获得下一个值: x = next(it) except StopIteration: # 遇到StopIteration就退出循环 break