Python:如何将成对扩展到三成？

Question

这个问题已经有答案了:

Rolling or sliding window iterator? (30 个回答)

已关闭 9 年前。

def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return izip(a, b)

考虑到这一点，你会如何得到:

s -> (s0,s1,s2), (s1,s2,s3), (s2,s3,s4), ...

Answer 1

def triwise(iterable):
    a, b = tee(iterable)
    a, c = tee(iterable)
    next(b, None)
    next(c, None)
    next(c, None)
    return izip(a, b, c)

应该做你想做的事。在某些时候，正确概括这一点可能是有意义的，但我将其作为练习留给读者:

>>> list(triwise(xrange(5)))
[(0, 1, 2), (1, 2, 3), (2, 3, 4)]