两个不同长度数组的和

Question

例如，我有两个数组

arraya[1,1,1,1,1,1,1]
arrayb[0,1,2]

我想把arrayb连续添加到arraya中，如下所示：

arraysum[1,2,3,1,2,3,1]

我该怎么做？你知道吗

Answer 1

arraya = [1,1,1,1,1,1,1]
arrayb = [0,1,2]
arraysum = []

i=0

while i in range(len(arraya)):
  arraysum.append(arraya[i] + arrayb[i % len(arrayb)])
  i+=1

print (arraysum)

Answer 2

您可以将zip与cycle结合使用：

if arrayb:
    arraysum = [sum(x) for x in zip(cycle(arrayb), arraya)]
else:
    arraysum = arraya

Answer 3

arraya = [1,1,1,1,1,1,1]
arrayb = [0,1,2]

for i in range(len(arraya)):
    arraya[i] += arrayb[i % len(arrayb)]

print arraya

产生 [1, 2, 3, 1, 2, 3, 1]