Python 遍历类的实例

在 Python 中，我们可以通过 for 循环遍历任何可迭代对象，包括类的实例。但是，当我们尝试遍历一个类的实例时，Python 解释器实际上会遍历实例的属性，而不是类本身。因此，如果我们想要遍历一个类的实例，我们需要提供一个迭代器方法。

实现一个迭代器方法

我们可以使用 Python 中的 __iter__ 和 __next__ 方法来定义实例的迭代器。__iter__ 方法应该返回一个迭代器对象，而 __next__ 方法则应该返回迭代器中的下一个元素。如果迭代器中没有更多元素，则应该引发 StopIteration 异常。

以下是一个示例，演示如何为一个简单的类实现迭代器方法：

class MyNumbers:
  def __init__(self, start, end):
    self.value = start
    self.end = end

  def __iter__(self):
    return self

  def __next__(self):
    if self.value > self.end:
      raise StopIteration
    current = self.value
    self.value += 1
    return current

让我们看看这个示例代码是如何工作的：

my_numbers = MyNumbers(1, 5)
for num in my_numbers:
  print(num)

输出:

1
2
3
4
5

改进迭代器方法

如果我们想要在返回每个元素之前对其进行转换或处理，则可以改进上面的迭代器方法。例如，我们可以将返回的元素加倍。以下是这个改进的示例：

class MyNumbers:
  def __init__(self, start, end):
    self.value = start
    self.end = end

  def __iter__(self):
    return self

  def __next__(self):
    if self.value > self.end:
      raise StopIteration
    current = self.value
    self.value += 1
    return current * 2

让我们再次运行代码并查看输出：

my_numbers = MyNumbers(1, 5)
for num in my_numbers:
  print(num)

输出：

2
4
6
8
10

结论

在 Python 中遍历类的实例非常容易，我们只需要为实例定义一个迭代器方法，然后使用 for 循环遍历该实例即可。如果我们想要在迭代过程中进行转换或处理，我们可以改进迭代器方法以满足我们的需求。