python list next_在Python中将列表迭代为pair(current,next)

这里有一个来自

itertools模块docs的相关示例:

import itertools

def pairwise(iterable):

"s -> (s0,s1), (s1,s2), (s2, s3), ..."

a, b = itertools.tee(iterable)

next(b, None)

return itertools.izip(a, b)

工作原理:

首先,创建两个并行迭代器a和b(tee()调用),两者都指向原始可迭代的第一个元素。第二个迭代器b向前移动一步(下一个(b,None))调用)。此时a点指向s0和b指向s1。 a和b都可以独立遍历原始迭代器 – izip函数接受两个迭代器,并使返回的元素成对,以相同的速度推进两个迭代器。

一个警告:tee()函数产生两个迭代器,它们可以彼此独立地前进,但这是有代价的。如果一个迭代器比另一个迭代器进一步前进,那么tee()需要在内存中保留消费的元素,直到第二个迭代器也将它们存储起来(它不能“回绕”原始迭代器)。这里没有关系,因为一个迭代器只比其他迭代器提前一步,但是一般来说,这样很容易使用大量的内存。

并且由于tee()可以取n个参数,这也可以用于两个以上的并行迭代器:

def threes(iterator):

"s -> (s0,s1,s2), (s1,s2,s3), (s2, s3,4), ..."

a, b, c = itertools.tee(iterator, 3)

next(b, None)

next(c, None)

next(c, None)

return itertools.izip(a, b, c)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值