### 代码概述 以下Python代码实现了基于提供的二维热传导方程参数识别与温度分布预测的任务。此代码首先读取已有的100秒内温度数据，然后通过有限差分法求解热传导方程，最后预测第500秒时的温度分布。 ### 代码解析 ```python import numpy as np from scipy.optimize import least_squares import matplotlib.pyplot as plt # 加载数据 data = np.load('heat_equation_data.npz') times = data['times'] # 时间点 u_obs = data['u'] # 观测到的温度值 def heat_eqn_residual(params, u, times): alpha, beta = params nx, ny, nt = u.shape dx, dy = 1/(nx-1), 1/(ny-1) dt = (times[-1] - times[0]) / (nt - 1) residuals = [] for k in range(1, nt-1): for i in range(1, nx-1): for j in range(1, ny-1): du_dt = (u[i, j, k+1] - u[i, j, k-1]) / (2*dt) laplacian = (u[i+1, j, k] - 2*u[i, j, k] + u[i-1, j, k]) / dx**2 \ + (u[i, j+1, k] - 2*u[i, j, k] + u[i, j-1, k]) / dy**2 source_term = 0.1 * np.sin(2*i*dx) * np.cos(2*j*dy) * np.exp(-0.01*times[k]) residual = du_dt - alpha * laplacian - beta * u[i, j, k] - source_term residuals.append(residual) return np.array(residuals) # 初始猜测值 initial_guess = [0.1, 0.1] # 使用最小二乘法优化参数 result = least_squares(heat_eqn_residual, initial_guess, args=(u_obs, times)) alpha_est, beta_est = result.x print(f'Estimated parameters: alpha={alpha_est:.4f}, beta={beta_est:.4f}') # 预测500秒后的温度分布 def predict_temperature(alpha, beta, T_pred=500, nx=50, ny=50, nt=5000): u = np.zeros((nx, ny, nt)) x = np.linspace(0, 1, nx) y = np.linspace(0, 1, ny) t = np.linspace(0, T_pred, nt) dx, dy = 1/(nx-1), 1/(ny-1) dt = T_pred / (nt - 1) # 设置初始条件 for i in range(nx): for j in range(ny): u[i, j, 0] = np.sin(np.pi*x[i]) * np.sin(np.pi*y[j]) # 显式有限差分法求解 for k in range(1, nt): for i in range(1, nx-1): for j in range(1, ny-1): u[i, j, k] = u[i, j, k-1] + dt * ( alpha * ((u[i+1, j, k-1] - 2*u[i, j, k-1] + u[i-1, j, k-1]) / dx**2 + (u[i, j+1, k-1] - 2*u[i, j, k-1] + u[i, j-1, k-1]) / dy**2) + beta * u[i, j, k-1] + 0.1 * np.sin(2*x[i]) * np.cos(2*y[j]) * np.exp(-0.01*t[k-1]) ) return x, y, t, u x_pred, y_pred, t_pred, u_pred = predict_temperature(alpha_est, beta_est) # 可视化结果 X, Y = np.meshgrid(x_pred, y_pred) plt.figure(figsize=(8,6)) plt.contourf(X, Y, u_pred[:,:, -1], cmap='hot') plt.colorbar(label='Temperature') plt.title('Predicted Temperature Distribution at 500s') plt.xlabel('x') plt.ylabel('y') plt.show() ``` ### 知识点 1. **有限差分法**：数值求解偏微分方程的方法之一。 2. **最小二乘法**：用于估计最佳拟合参数的一种优化算法。 3. **Matplotlib绘图**：可视化计算结果，帮助理解数据。