在Python中以对（current，next）的形式迭代列表

一尘不染

在Python中以对（current，next）的形式迭代列表

python

有时我需要在Python中迭代一个列表，以查看“当前”元素和“下一个”元素。到目前为止，我已经使用以下代码完成了此操作：

for current, next in zip(the_list, the_list[1:]):
    # Do something

这行得通，符合我的期望，但是有没有一种更惯用或有效的方式来执行相同的操作？

阅读 1285

2020-02-11

共1个答案

一尘不染

这是itertools模块文档中的一个相关示例：

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

对于Python 2，你需要itertools.izip代替zip：

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向前移动了1个（next(b, None)）调用。此时指向as0并b指向s1。双方a并b可以独立遍历原来迭代器-的izip函数接受两个迭代器，使对返回的元素，以相同的速度前进的两个迭代器。

一个警告：该tee()函数产生两个可以彼此独立进行的迭代器，但这要付出一定的代价。如果一个迭代器比另一个迭代器前进得更多，则tee() 需要将消耗的元素保留在内存中，直到第二个迭代器也将它们包括在内（它无法“倒回”原始迭代器）。这里没有关系，因为一个迭代器仅比另一个迭代器领先1步，但是通常使用这种方式很容易使用大量内存。

并且由于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 zip(a, b, c)

2020-02-11