微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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.

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐