我如何创建一个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循环或一个列表推导式,以便每次迭代都给我两个元素?
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,v in enumerate(items[:-1]):
for u in items[i+1:]:
其他回答
您需要一个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)
这里我们有alt_elem方法,它可以放入for循环中。
def alt_elem(list, index=2):
for i, elem in enumerate(list, start=1):
if not i % index:
yield tuple(list[i-index:i])
a = range(10)
for index in [2, 3, 4]:
print("With index: {0}".format(index))
for i in alt_elem(a, index):
print(i)
输出:
With index: 2
(0, 1)
(2, 3)
(4, 5)
(6, 7)
(8, 9)
With index: 3
(0, 1, 2)
(3, 4, 5)
(6, 7, 8)
With index: 4
(0, 1, 2, 3)
(4, 5, 6, 7)
注意:考虑到在func中执行的操作,上述解决方案可能不是有效的。
可以使用more_itertools包。
import more_itertools
lst = range(1, 7)
for i, j in more_itertools.chunked(lst, 2):
print(f'{i} + {j} = {i+j}')
我认为这是一个分享我对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)
你需要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]表示创建元素的子集集合(索引% 2 == 0) Zip (x,y)从x和y集合中创建一个相同索引元素的元组集合。