如何用NumPy实现无循环的特定累积点积计算？

给定两个长度为n的向量a=(a₁,…,aₙ)和b=(b₁,…,bₙ)，需要构建向量c，其中每个元素cₖ满足：
cₖ = a₁bₙ₋ₖ₊₁ + a₂bₙ₋ₖ₊₂ + … + aₖbₙ

比如当a=[1,2,3,4]、b=[5,6,7,8]时，期望得到c=[8,23,44,70]。你提供的循环实现代码里注释结果有误，正确运行后会得到期望的目标值。

完全可以用NumPy实现无循环计算，最简洁高效的方式是利用卷积运算：

实现思路 #

这个运算本质等价于：将向量b反转后，与向量a做一维卷积，再截取前n个元素（n为向量长度）。

代码示例 #

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([5, 6, 7, 8])

# 反转b向量，将原b的后k个元素转为前k个
b_reversed = b[::-1]
# 计算全模式卷积，截取前len(a)个元素即为目标向量c
c = np.convolve(a, b_reversed)[:len(a)]

print(c)  # 输出: [ 8 23 44 70]

原理说明 #

np.convolve(a, b_reversed, mode='full')会计算两个向量所有可能的重叠点积：

• 计算c₁时，对应a的前1个元素与b_reversed的前1个元素（原b的最后1个元素）点积，得到8；
• 计算c₂时，对应a的前2个元素与b_reversed的前2个元素（原b的最后2个元素）点积，1×8+2×7=23；
  以此类推，卷积结果的前n个元素正好对应所有cₖ的值。

如果担心大向量场景下的内存问题，也可以用矩阵乘法实现，但效率不如卷积：

# 构造a的下三角矩阵，每行保留前k个元素（k从1到n）
a_tril = np.tril(np.tile(a, (len(a), 1)))
c = a_tril.dot(b_reversed)

内容的提问来源于stack exchange，提问作者Geodezik