假设我有一个数组 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)。
最佳答案
定义A_left = A[:-2, :]
、A_middle = A[1:-1, :]
和 A_right = A[2: ,:]
然后B = a*A_left + b*A_middle + c*A_right
关于python - Numpy 数组替代 for 循环,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33127245/