python - python为数组创建卷积函数

Question

我有两个数组，

a = [3, 6, 8, 2, 5, 5]
b = [2, 7, 9]

我需要创建一个新数组c来获取值并像这样添加它们：a[0+0]*b[0] + a[0+1]*b[1] + a[0+2]*b[2] = (3*2) + (6*7) + (9*8) = 6 + 42 + 72这意味着c[0] = 120

我完全不知道如何做到这一点，任何能将我指向正确方向的事情都会很棒。

Answer 1

total = 0
for n in range(0, min(len(a), len(b))):
    total += a[n] * b[n]

范围函数

Answer 2

如果c[k] = a[k+0]*b[0] + a[k+1]*b[1] + a[k+2]*b[2]

然后

>>> c = [sum(i*j for i,j in zip(a[k:], b)) for k in range(4)]
>>> c
[120, 86, 75, 84]

Answer 3

我认为这会做你想要的。它从@DukeSilver 的答案中借用了一些代码，并使其构建一个列表，而不仅仅是计算单个值。我的假设是a总是比b.

c = [sum(a[i+j]*b[j] for j in range(len(b))) for i in range(len(a) - len(b) + 1)]

Answer 4

numpy.convolve还是您想编写自己的函数？