python - 如何在不更改其预查询状态的情况下在python中查询迭代器

Question

我构建了一个包含其他对象 B 列表的可迭代对象 A。如果在 for 循环中使用对象 A 时它被标记为错误，我希望能够自动跳过列表中的特定对象 B。

class A():
  def __init__(self):
    self.Blist = [B(1), B(2), B(3)] #where B(2).is_bad() is True, while the other .is_bad() are False

  def __iter__(self):
    nextB = iter(self.Blist)
    #if nextB.next().is_bad():
    #   return after skip
    #else:
    #   return nextB

但是，我无法弄清楚如何在不跳过查询的迭代(else 子句失败)的情况下编写上面伪代码中注释的条件

谢谢!

Answer 1

您可以使用生成器函数:

  def __iter__(self):
      for item in self.Blist:
          if not item.is_bad():
              yield item

生成器函数由关键字yield 标记。生成器函数返回一个生成器对象，它是一个迭代器。它将在 yield 语句处暂停执行，然后在调用例程在 interator 上调用 next 时恢复处理。