如何组合2个向量?

时间:2014-05-07 17:11:15

标签: python list function

假设我有2个向量[2,4,6,8]和[1,3,5]。

如果我根据他们的索引组合数字,我希望[3,7,11]作为我的结果。如果一个向量的索引多于另一个向量,则它将在计算中被排除。 (这就是为什么8不包括在这里)。我的问题是如何组合2个向量而忽略额外的8,如上所示?我需要一个功能。

def v_add(num1, num2):
    total = num1 + num2
    return total

3 个答案:

答案 0 :(得分:5)

你没有矢量,但列表。

def v_add(num1, num2):
    return [a+b for a,b in zip(num1, num2)]

答案 1 :(得分:3)

a = [1, 2, 3]
b = [1, 2, 3, 4]
c = map(sum, zip(a, b))
print(c)

<强>输出

[2, 4, 6]

答案 2 :(得分:0)

a = [1,2,3]
b = [4,5,6]
print [a[i] + b[i] for i in range(min(len(a),len(b)))]