Python：如何断言可迭代的第一个元素的类型？ [复制]

Question

我正在寻找一种“pythonic”方式来断言迭代中第一个元素的类型。

例如，如果我有以下列表：

l = [(1,2,3), (4,5,6), ...]

它传入的函数，我可以简单地做：

def foo(l):
    assert isinstance(l[0], tuple)
    ...

现在我正在寻找类似的东西，它也适用于迭代器，但不会将整个列表加载到 ram 中。我可以试试：

def foo(it):
    assert isinstance(next(it), tuple)
    ...

it = iter(l)
foo(it)

但这显然会修改断言中迭代器的状态。

有没有一种简单的方法可以在不修改迭代器的第一个元素的情况下查看它？

Answer 1

我目前（我觉得很丑）的解决方案是：

from itertools import tee

def foo(it):
    if __debug__:
        it1, it2 = tee(it, 2)
        assert isinstance(next(it1), tuple)
        it = it2
    ...

有没有更好/更酷的方法来查看我不知道的第一个元素？