检测迭代器是否将被消耗

Question

有没有一种统一的方法可以知道迭代是否会消耗iterable对象？在

假设您有一个函数crunch，它请求一个iterable对象作为参数，并多次使用它。比如：

def crunch (vals):

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

（注意：合并两个for循环不是一个选项）。在

如果使用不是列表的iterable调用函数，则会出现问题。在下面的调用中，yum函数从不执行：

^{pr2}$

原则上，我们可以通过如下重新定义crunch函数来解决此问题：

def crunch (vals):
    vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

但是如果对crunch的调用是：

hugeList = list(longDataStream)
crunch(hugeList)

我们可以通过如下定义crunch来解决此问题：

def crunch (vals):
    if type(vals) is not list:
        vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

但是仍然存在调用代码将数据存储在

• 无法消费
• 不是列表

例如：

from collections import deque
hugeDeque = deque(longDataStream)
crunch(hugeDeque)

最好有一个isconsumable谓词，这样我们就可以像这样定义crunch：

def crunch (vals):
    if isconsumable(vals):
        vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

这个问题有解决办法吗？在

Answer 1

def crunch (vals):
    vals1, vals2 = itertools.tee(vals, 2)

    for v in vals1:
        chomp(v)

    for v in vals2:
        yum(v)

在这种情况下，tee将在内部存储{}的完整性，因为一个迭代器在另一个迭代器启动之前完成