如何求和两个列表N元素

Question

我需要添加两个数字列表.当一个列表大于另一个列表时,我得到错误"IndexError:list index out of range".错误发生在行resultado = v1[i] + v2[i],但我不知道如何解决它.

我的代码是:

def suma(A, B):

    if len(A) > len(B):
        v1 = A
        v2 = B
    else:
        v1 = B
        v2 = A
    v3 = []
    i = 0
    for i in range(len(v1)):
        resultado = v1[i] + v2[i]
        v3.append(resultado)
    for j in range(len(v1), len(v2)):
        v3.append(v2[j])
    return v3

vectorA = []
vectorB = []
x = 1
while x !=0:
    print "Ingrese los datos para el primer vector (0 para finalizar)"
    x = input("-->>")
    if x !=0:
        vectorA.append (x)

print "=" * 30
x = 1
while x !=0:
    print "Ingrese los datos para el segundo vector (0 para finalizar)"
    x = input("-->>")
    if x !=0:
        vectorB.append (x)

print "=" * 30
print vectorA
print vectorB
print "=" * 30
print "A) SUMA DE VECTORES"
print "=" * 30

suma(vectorA, vectorB)
print suma

Answer 1

>>> import itertools
>>> A = [1, 2, 3]
>>> B = [10, 11]
>>> [a + b for a, b in itertools.izip_longest(A, B, fillvalue=0)]
[11, 13, 3]

• itertools.izip_longest

所以,

def suma(A, B):
    return [a + b for a, b in itertools.izip_longest(A, B, fillvalue=0)]