目标:例如给定有限迭代器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).
如何有效地实现这个附加功能?
解决方法:
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.