Python中,迭代器(Iterator)是一个对象,它实现了迭代器协议,即包含 __iter__() 方法和 __next__() 方法(在Python 2中为 next() 方法)。迭代器可以用于遍历容器对象(如列表、元组、字典等)中的元素,或者用于生成无限序列的数据。本文主要介绍一下Python中迭代器(Iterator)。

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

推荐文档