如何从python中的两个不同列表计算元素?

时间:2011-05-11 20:35:54

标签: python list variables object

我有两个整数列表,我想系统地操作它们的对象。例如,我有:

A = [ a1, a2, a3, a4 ...]
B = [ b1, b2, b3 ...]

我希望 打印 这样的事情:

a1+b1   a2
a2+b2   a3
a3+b3   a4

我认为有一种“ 循环”方式,但我不知道如何在“ 中使用两个变量 循环“。

6 个答案:

答案 0 :(得分:3)

您可以使用zip

>>> A = ['a1', 'a2', 'a3', 'a4']
>>> B = ['b1', 'b2', 'b3']
>>> zip(A[:3], B, A[1:])
[('a1', 'b1', 'a2'), ('a2', 'b2', 'a3'), ('a3', 'b3', 'a4')]
>>> for a, b, c in zip(A[:3], B, A[1:]):
...     print a + '+' + b + '   ' + c
... 
a1+b1   a2
a2+b2   a3
a3+b3   a4

答案 1 :(得分:2)

你在寻找像这样简单的东西:

In []: A= [1, 2, 3, 4]
In []: B= [1, 2, 3]
In []: for k, b in enumerate(B):
   ..:     print A[k]+ b, A[k+ 1]
   ..:
2 2
4 3
6 4

也许是这样的:

In []: for k, b in enumerate(B):
   ..:     print '{}+{}\t{}'.format(A[k], b, A[k+ 1])
   ..:
1+1     2
2+2     3
3+3     4

答案 2 :(得分:2)

使用pairwise recipe from itertools

from itertools import tee, izip

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

a = [1,2,3,4]
b = [7,8,9]

for (a1, a2), b1 in izip(pairwise(a), b):
    print a1 + b1, a2

答案 3 :(得分:1)

for i,b in enumerate(B):
    print(A[i] + '+' + b + '\t' + A[i+1])

答案 4 :(得分:1)

[(a + b, c) for (a, b, c) in zip(A, B, A[1:])]

无论A和B包含什么

答案 5 :(得分:1)

这一个

l = zip(map(sum, zip(A, B)), A[1:])

将生成

的元组列表
[(a_1 + b_1, a_2), (a_2 + b_2, a_2), ..., (a_(i-1) + b_(i-1), a_i)]

要打印,

for a, b in l:
    print a, b