1.png
3.png
import numpy as np
import matplotlib.pyplot as plt
nx = 41 # 网格节点数
dx = 2/(nx - 1) # 网格尺寸
nt = 40 # 时间步数
nu = 0.3 # 扩散系数
sigma = 0.2 #定义一个中间变量，用于确定时间步长dt
dt = sigma * dx**2 / nu #时间步长
# 定义初始条件
u = np.ones(nx)
u[int(0.5 / dx):int(1 / dx + 1)] = 2
un = np.ones(nx) # 定义一个定义数组
plt.ion()
plt.show()
for n in range(nt): #时间循环，计算每一个时间点上网格点上的数据
plt.cla()
un = u.copy()
for i in range(1,nx-1): #空间循环，计算u[i]
u[i] = un[i]+ nu *dt /dx**2 * (un[i+1] -2 * un[i] + un[i-1])
# 下面为图形绘制代码
plt.plot(np.linspace(0,2,nx) , u , 'm' , lw =3 ) #绘制速度随时间变化曲线
plt.ylim(0.9,2.1) # 确定y轴上下限
plt.title("times: %.4f s"% (n*dt)) # 将时间打印到图形上
plt.pause(0.1) #控制图形显示时间
plt.ioff()
plt.show()
系列教程见如下链接：
https://mp.weixin.qq.com/s/2nFR7efdfnX3VPRdskOeng
