python - 排除最后一个元素的迭代器

Question

什么是在不知道最后一个元素长度的情况下实现排除最后一个元素的迭代器的 pythonic 方法？

一个例子:

>>> list(one_behind(iter(range(10)))
... [0, 1, 2, 3, 4, 5, 6, 7, 8]

>>> iter_ = one_behind(iter((3, 2, 1)))
>>> next(iter_)
... 3
>>> next(iter_)
... 2
>>> next(iter_)
... StopIteration

一个简单的方法是使用循环并存储先前的值，但我想要更短的东西。

使用循环的引用实现:

def one_behind(iter_):
    prev = None
    for i, x in enumerate(iter_):
        if i > 0:
            yield prev
        prev = x

Answer 1

使用 itertools.tee :

import itertools

def behind(it):
    # it = iter(it)  # to handle non-iterator iterable.
    i1, i2 = itertools.tee(it)
    next(i1)
    return (next(i2) for x in i1)

用法:

>>> list(behind(iter(range(3))))
[0, 1]