python - Numpy 数组替代 for 循环

Question

假设我有一个数组 A[i,j]，我想通过执行以下操作来计算新的数量 B[i,j]:

for i in arange(1, n):
    B[i,j+1] = a*A[i-1,j] + b*A[i,j]+ c*A[i+1,j]

我可以使用什么 numpy 功能将其转换为纯数组操作？

我正在考虑 np.cumprod 但不清楚如何将系数纳入计算中。

A.shape 将为 (n+1, j)。

Answer 1

定义A_left = A[:-2, :]、A_middle = A[1:-1, :] 和 A_right = A[2: ，:]

然后B = a*A_left + b*A_middle + c*A_right