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

发布时间：2020-12-20 11:48:00 所属栏目：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(itera

目标：例如给定有限迭代器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.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!