`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")`