Python 迭代器

Python 迭代器

迭代器是一种对象，该对象包含值的可计数数字。

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

从技术上讲，在 Python 中，迭代器是实现迭代器协议的对象，它包含方法 __iter__() 和 __next__()。

迭代器 VS 可迭代对象（Iterable）

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

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

Eksempel

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

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

Kør Eksempel

甚至连字符串都是可迭代的对象，并且可以返回迭代器：

Eksempel

字符串也是可迭代的对象，包含一系列字符：

mystr = "banana"
myit = iter(mystr)
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))

Kør Eksempel

遍历迭代器

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

Eksempel

迭代元组的值：

mytuple = ("apple", "banana", "cherry")
for x in mytuple:
  print(x)

Kør Eksempel

Eksempel

迭代字符串中的字符：

mystr = "banana"
for x in mystr:
  print(x)

Kør Eksempel

提示：for 循环实际上创建了一个迭代器对象，并为每个循环执行 next() 方法。

创建迭代器

要把对象/类创建为迭代器，必须为对象实现 __iter__() 和 __next__() 方法。

正如您在 Python 类/对象 一章中学到的，所有类都有名为 __init__() 函数，它允许您在创建对象时进行一些初始化。

__iter__() Metoden fungerer på samme måde, og du kan udføre operationer (initialisering osv.), men skal altid returnere iteratorobjektet selv.

__next__() Metoden tillader også, at du udfører operationer, og skal altid returnere det næste element i sekvensen.

Eksempel

Opret en iterator, der returnerer tal, starter fra 1 og øger hver sekvens med 1 (returnerer 1, 2, 3, 4, 5 osv.):

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))

Kør Eksempel

StopIteration

Hvis du har nok next() sætning, eller i en for-løkke, vil ovenstående eksempel køre for evigt.

For at undgå, at iterationen bliver ved for evigt, kan vi bruge StopIteration sætning.

i __next__() I metoder, hvor vi udfører en bestemt antal iterationer, kan vi tilføje en stopbetingelse for at udløse en fejl:

Eksempel

Stop efter 20 iterationer:

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)

Kør Eksempel