Python 迭代器

Python迭代器

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

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

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

Iterator 和 Iterable

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

所有这些对象都有一个获取获取迭代器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)

for循环实际创建了迭代器对象，并且在每个循环中执行next()方法。

创建一个迭代器

要创建一个对象或类作为迭代器，您必须实现方法 __iter__()和 __next__()对象。

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

__iter__()方法行为类似，您可以执行操作（初始化等），但必须始终返回迭代器对象本身。

__next__()方法还允许您执行操作，并且必须返回序列中的下一个项目。

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__()方法中，如果迭代完成指定的次数，我们可以添加一个终止条件来引发错误：

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)