所以我试图在Python中将以下等式实现为2D数组:
u [i] [j + 1] = u [i] [j] + dt * l *(u [i-1] [j] - u [i] [j])。这是我必须这样做的代码:
l = 0.3n = 3m = 100dt = 3.0T = ……
如果你改变了 c='r' 成 c='r-' 然后matplotlib试图解释 r- 作为一种颜色,这就是它会给出错误的原因。
c='r'
c='r-'
r-
我不是100%肯定你想要的结果,但我认为你可以通过定义完全删除循环的绘图 j=np.arange(m-1) :
j=np.arange(m-1)
# The rest of your code remains the same j = np.arange(m-1) plt.plot(t, u01, 'g-') plt.plot(j, u[1][j + 1], 'r-') plt.plot(j, u[2][j + 1], 'b-') plt.show()
这给了我: