数学建模习题8.5

`import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import solve_ivp

定义微分方程模型

def model(t, y):
f, df_dm, d2f_dm2, T, dT_dm = y
d3f_dm3 = -3 * f * d2f_dm2 + 2 * (df_dm)**2 - T
d2T_dm2 = -2 * f * dT_dm
return [df_dm, d2f_dm2, d3f_dm3, dT_dm, d2T_dm2]

初始条件

y0 = [0, 0, 0.68, 1, -0.5]
t_span = (0, 10)
t_eval = np.linspace(t_span[0], t_span[1], 1000)

解微分方程

sol = solve_ivp(model, t_span, y0, t_eval=t_eval, method='RK45')

提取解

f = sol.y[0]
T = sol.y[3]

绘制 f(m) 和 T(m) 的图像

plt.figure(figsize=(12, 6))
plt.subplot(2, 1, 1)
plt.plot(sol.t, f, label='f(m)')
plt.xlabel('m')
plt.ylabel('f(m)')
plt.title('Solution for f(m)')
plt.grid(True)

plt.subplot(2, 1, 2)
plt.plot(sol.t, T, label='T(m)', color='orange')
plt.xlabel('m')
plt.ylabel('T(m)')
plt.title('Solution for T(m)')
plt.grid(True)

plt.tight_layout()
plt.show()

print("学号后四位:05")`

posted on 2024-11-17 15:08  VVV1  阅读(1)  评论(0编辑  收藏  举报