我如何创建一个for循环或一个列表推导式,以便每次迭代都给我两个元素?

l = [1,2,3,4,5,6]

for i,k in ???:
    print str(i), '+', str(k), '=', str(i+k)

输出:

1+2=3
3+4=7
5+6=11

当前回答

for (i, k) in zip(l[::2], l[1::2]):
    print i, "+", k, "=", i+k

Zip (*iterable)返回一个元组,其中包含每个迭代对象的下一个元素。

L[::2]返回列表的第1、3、5等元素:第一个冒号表示切片从开头开始,因为后面没有数字,第二个冒号只在你想要“切片中的步骤”(在本例中是2)时才需要。

L[1::2]做同样的事情,但从列表的第二个元素开始,因此它返回原始列表的第2、第4、第6等元素。

其他回答

我认为这是一个分享我对n>2的概括的好地方,它只是一个可迭代对象上的滑动窗口:

def sliding_window(iterable, n):
    its = [ itertools.islice(iter, i, None) 
            for i, iter
            in enumerate(itertools.tee(iterable, n)) ]                               

    return itertools.izip(*its)

这个问题的标题有误导性,你似乎在寻找连续的对,但如果你想遍历所有可能的对的集合,那么这样做是可行的:

for i,v in enumerate(items[:-1]):
        for u in items[i+1:]:

虽然使用zip的所有答案都是正确的,但我发现自己实现功能会导致更可读的代码:

def pairwise(it):
    it = iter(it)
    while True:
        try:
            yield next(it), next(it)
        except StopIteration:
            # no more elements in the iterator
            return

it = iter(it)部分确保它实际上是一个迭代器,而不仅仅是一个可迭代对象。如果它已经是一个迭代器,这一行是一个无操作。

用法:

for a, b in pairwise([0, 1, 2, 3, 4, 5]):
    print(a + b)

您需要一个pairwise()(或grouped())实现。

def pairwise(iterable):
    "s -> (s0, s1), (s2, s3), (s4, s5), ..."
    a = iter(iterable)
    return zip(a, a)

for x, y in pairwise(l):
   print("%d + %d = %d" % (x, y, x + y))

或者,更一般地说:

def grouped(iterable, n):
    "s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), (s2n,s2n+1,s2n+2,...s3n-1), ..."
    return zip(*[iter(iterable)]*n)

for x, y in grouped(l, 2):
   print("%d + %d = %d" % (x, y, x + y))

在Python 2中,您应该导入izip来替换Python 3的内置zip()函数。

这都归功于martineau对我问题的回答,我发现这是非常有效的,因为它只在列表上迭代一次,并且不会在这个过程中创建任何不必要的列表。

注意:不要将其与Python自己的itertools文档中的成对recipe混淆,后者生成s -> (s0, s1), (s1, s2), (s2, s3),…,正如@lazyr在评论中指出的那样。

对于那些想在Python 3上使用mypy进行类型检查的人来说,这是一个小补充:

from typing import Iterable, Tuple, TypeVar

T = TypeVar("T")

def grouped(iterable: Iterable[T], n=2) -> Iterable[Tuple[T, ...]]:
    """s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), ..."""
    return zip(*[iter(iterable)] * n)
for (i, k) in zip(l[::2], l[1::2]):
    print i, "+", k, "=", i+k

Zip (*iterable)返回一个元组,其中包含每个迭代对象的下一个元素。

L[::2]返回列表的第1、3、5等元素:第一个冒号表示切片从开头开始,因为后面没有数字,第二个冒号只在你想要“切片中的步骤”(在本例中是2)时才需要。

L[1::2]做同样的事情,但从列表的第二个元素开始,因此它返回原始列表的第2、第4、第6等元素。