python – 最后得到具有附加项的成对迭代器

目标：例如给定有限迭代器p0,p1,…,pn变为(p0,p1),(p1,p2),(pn-1,pn),(pn,无) – 迭代器通过成对的连续项特别的最后一项.

pairwise()函数作为itertools用法的示例存在于文档中：

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

但我还想在迭代器的末尾添加另一个项目(如果它是有限的),对于第二个元素的一些默认值(例如,None).

如何有效地实现这个附加功能？

解决方法

使用 itertools.zip_longest：

def pairwise(iterable):
    "s -> (s0,None)
    return zip_longest(a,b)

当其中一个输入迭代器用完时,zip_longest会使用填充值填充它,默认值为None.