最后获得成对迭代器和附加项

Question

目标:例如,给定有限迭代器p0, p1, ..., pn转换为(p0, p1), (p1, p2), ..., (pn-1, pn), (pn, None)- 迭代器通过成对的具有特殊最后项的连续项.

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

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

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

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

Answer 1

使用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.