生成的2D矩阵的

Question

我为我的学校项目编写了代码。 u和v矩阵已经生成。我已经制作了2D轮廓，但我希望用3D绘制u和v。

import numpy as np
#import matplotlib.pyplot as plt
#inputs
m = 40
n = 20
Uinf = 2
delX = 0.05
delY = 0.05
nu = 0.8e-2
u= np.zeros((m+1,n))
v= np.zeros((m+1,n))
#intializing boundry conditions
u[:,0] = v[:,0] = v[:,n-1] = v[0,:] = 0
u[:,n-1] = u[0,:] = Uinf
#main program
for i in range(0,m):
    for j in range(1,n-1):
        a = (nu*delX) / (u[i,j]*delY**2)
        b = (v[i,j]*delX) / (2*u[i,j]*delY)
        u[i+1,j] = (a-b)*u[i,j+1]+(1-2*a)*u[i,j]+(a+b)*u[i,j-1]
        v[i+1,j] = v[i+1,j-1]-(delY/(2*delX))*(u[i+1,j]-u[i,j]+u[i+1,j-1]-u[i,j-1])

我必须提到我不想生成任意函数; u [i，j]和v [i，j]存在。 谢谢！