python - 发电机输出长度

Question

这个问题在这里已经有了答案:

Getting number of elements in an iterator in Python (19 个回答)

What's the shortest way to count the number of items in a generator/iterator? (7 个回答)

关闭 4年前。

Python 提供了一种很好的方法来获取急切可迭代的长度，即 len(x)。但是对于由生成器理解和函数表示的惰性迭代，我找不到任何类似的东西。当然，这样写也不难:

def iterlen(x):
  n = 0
  try:
    while True:
      next(x)
      n += 1
  except StopIteration: pass
  return n

但我无法摆脱我正在重新实现自行车的感觉。

(当我输入函数时，我突然想到一个想法:也许真的没有这样的函数，因为它“破坏”了它的论点。不过，这对我来说不是问题)。

P.S.:关于第一个答案 - 是的，像 len(list(x)) 这样的东西也可以，但这会大大增加内存的使用。

P.P.S.:重新检查...忽略 P.S.，似乎我在尝试时犯了一个错误，它工作正常。给您添麻烦了。

Answer 1

最简单的方法可能只是 sum(1 for _ in gen) 其中 gen 是您的生成器。