优化嵌套的for循环

我正在尝试避免使用for循环来运行我的计算。但是我不知道该怎么做。我有一个w形状矩阵(40,100)。每条线一次都保持波浪的位置t。例如，第一行w[0]是初始条件（也是w[1]出于我将要说明的原因）。

为了计算下一行元件我使用，对于每一个t和x上形状范围：

w[t+1,x] = a * w[t,x] + b * ( w[t,x-1] + w[t,x+1] ) - w[t-1,x]

其中a和b是基于方程解的一些常量（它其实并不重要）， ，a = 2(1-r)，。b=r r=(c*(dt/dx))**2哪里c的波速和dt，与方向和dx增量有关。xt

有什么办法可以避免for循环：

for t in range(1,nt-1):
    for x in range(1,nx-1):
      w[t+1,x] = a * w[t,x] + b * ( w[t,x-1] + w[t,x+1] ) - w[t-1,x]

nt和nx是w矩阵的形状。