Python 迭代器

迭代器是一个包含可数个值的对象。

迭代器是一个可以迭代的对象，这意味着您可以遍历所有值。

从技术上讲，在Python中，迭代器是一个实现迭代器协议的对象，它由以下方法组成__iter__()和__next__()。

迭代器与可迭代器

列表、元组、字典和集合都是可迭代对象。它们是可迭代的容器您可以从中获取迭代器。

所有这些对象都有一个iter()用于获取迭代器的方法：

示例

从元组返回一个迭代器，并打印每个值：

mytuple = ("apple", "banana", "cherry")
myit = iter(mytuple)

print(next(myit))
print(next(myit))
print(next(myit))

亲自试一试 »

即使字符串也是可迭代对象，并且可以返回迭代器：

示例

字符串也是可迭代对象，包含字符序列：

mystr = "banana"
myit = iter(mystr)

print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))

亲自试一试 »

循环遍历迭代器

我们还可以使用一个for循环遍历可迭代对象：

示例

迭代元组的值：

mytuple = ("apple", "banana", "cherry")

for x in mytuple:
print(x)

亲自试一试 »

示例

迭代字符串的字符：

mystr = "banana"

for x in mystr:
print(x)

亲自试一试 »

这个forLoop 实际上创建一个迭代器对象并为每个循环执行 next() 方法。

创建迭代器

要创建一个对象/类作为迭代器，您必须实现这些方法__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))

亲自试一试 »

停止迭代

如果您有足够的 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)

亲自试一试 »