如何记录迭代器产生 n 次或零次？ [复制]

Question

我想记录迭代器中事物的数量（而不是试图捕获一个空的迭代器，例如 this question），在我对它进行迭代并为每件事做了其他事情之后。

我的直觉是使用内置的enumerate：

def foo_yielder():
     to_yield = ['foo', 'bar']
     for x in to_yield:
         yield x

for i, x in enumerate(foo_yielder(), start=1):
    print(x)

# Now log how many things we printed
print(f"We logged {i} things")

但如果迭代器没有任何可以产生的东西，i 永远不会被定义。

如果这是 Javascript，我会这样做：

`We logged ${i || 0} things`

从 C 中吸取教训，我们可以做到：

i = 0
for x in foo_yielder():
    print(x)
    i = i + 1
print("We logged " + i + " things.")

但这似乎令人难以置信的蹩脚。

我可以在这里做一件优雅的事情吗？

Answer 1

最简单的方法是在enumerate 循环之前添加一个i = 0。

---

如果麻烦的是 i = 0 附加行很难看，您可以尝试以下方法：

添加一个 "dummy" 迭代器并将 chain 它添加到您的迭代器中。然后，将enumerate 重置为从0 开始计数（作为默认值）。这样，您可以确保enumerate 将迭代至少一项（并将i 设置为0），然后从1 计算实际迭代器：

from itertools import chain

for i, x in enumerate(chain([0], foo_yielder())):
    pass
print(f"We logged {i} things")

这会带来另一个问题，即x 在第一次迭代中始终是 dummy。您可以使用循环内的条件来处理它，但这会使代码更加复杂。

---

最后，您可以用try/except 包装最终打印：

try:
    print(f"We logged {i} things")
except NameError:
    print(f"We logged 0 things")

但同样，这只是更多的代码，而不是简单的i = 0。

Answer 2

将其设置为枚举空列表，for 循环上的else 子句将始终在循环完成后执行，检查当前命名空间中是否存在i。插页式的 print 声明并不是真正需要的。我不知道它是否完全优雅，但它是 for-else 成语的一个很好的用例。

for i, _ in enumerate([], start=1):
    pass # do whatever you want here
else:
    if "i" in dir():
        print(f"Found {i} instances")
    else:
        print(f"Found instances")

当然，这假设您没有在其他地方使用 i 作为变量

当然，稍微干净一点（干，等等）

for i, _ in enumerate([], start=1):
    pass # do whatever you want here
else:
    if "i" not in dir():
        i = 0
print(f"We logged {i} things")