遍历列表中的每两个元素

新手上路,请多包涵

我如何制作 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

原文由 jackhab 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 972
2 个回答

您需要一个 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我的问题 的回答,我发现这是非常有效的,因为它只在列表上迭代一次,并且不会在此过程中创建任何不必要的列表。

注意:这不应该与 pairwise Python 自己的 itertools 文档 中的配方混淆,它产生 s -> (s0, s1), (s1, s2), (s2, s3), ... 评论中指出的@as.lazy-

对于那些想在 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)

原文由 johnsyweb 发布,翻译遵循 CC BY-SA 4.0 许可协议

那么你需要 2 个元素的元组,所以

data = [1,2,3,4,5,6]
for i,k in zip(data[0::2], data[1::2]):
    print str(i), '+', str(k), '=', str(i+k)

在哪里:

  • data[0::2] 表示创建元素的子集 (index % 2 == 0)
  • zip(x,y) 从 x 和 y 集合相同的索引元素创建元组集合。

原文由 Margus 发布,翻译遵循 CC BY-SA 2.5 许可协议

推荐问题