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

2023年3月20日 205次阅读

目标：例如给定有限迭代器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,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return zip_longest(a, b)

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

点赞