迭代列表中的所有连续项对

Question

给出一个清单

l = [1, 7, 3, 5]

我想迭代所有连续列表项对(1,7), (7,3), (3,5),即

for i in xrange(len(l) - 1):
    x = l[i]
    y = l[i + 1]
    # do something

我想以更紧凑的方式做到这一点,比如

for x, y in someiterator(l): ...

有没有办法使用内置的Python迭代器来做到这一点？我确定该itertools模块应该有一个解决方案,但我无法弄明白.

Answer 1

只需使用拉链

>>> l = [1, 7, 3, 5]
>>> for first, second in zip(l, l[1:]):
...     print first, second
...
1 7
7 3
3 5

根据建议,您可以考虑izip在itertools非常长的列表中使用该函数,而不想创建新列表.

import itertools

for first, second in itertools.izip(l, l[1:]):
    ...

Answer 2

看一下pairwiseitertools的食谱:http://docs.python.org/2/library/itertools.html#recipes

引自那里:

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

一般版本

产生任何给定正自然大小的元组的通用版本可能如下所示:

def nwise(iterable, n=2):                                                      
    iters = tee(iterable, n)                                                     
    for i, it in enumerate(iters):                                               
        next(islice(it, i, i), None)                                               
    return izip(*iters)   

Answer 3

我会grouper像这样创建一个通用生成器

def grouper(input_list, n = 2):
    for i in xrange(len(input_list) - (n - 1)):
        yield input_list[i:i+n]

样品运行1

for first, second in grouper([1, 7, 3, 5, 6, 8], 2):
    print first, second

产量

1 7
7 3
3 5
5 6
6 8

样品运行1

for first, second, third in grouper([1, 7, 3, 5, 6, 8], 3):
    print first, second, third

产量

1 7 3
7 3 5
3 5 6
5 6 8