python - 如何在python中实现 "cumdot"

Question

假设我有两个长度相同的一维数组(即n)，现在我想实现一个“cumdot”函数，它输出长度为n的一维数组，并且可以用纯python代码实现

def cumdot(a,b):#a,b are two 1d arrays with same length
    n = len(a)
    output = np.empty(n)
    for i in range(n):
        output[i] = np.dot(a[:i+1],b[:i+1])
    return output

如何更有效地实现“cumdot”功能？

Answer 1

def cumdot(a, b):
    return numpy.cumsum(a * b)