Matlab代码的Python实现--有限差分法

% Matlab代码示例
L = 1; % 区间长度
N = 100; % 网格点数
dx = L / N; % 网格步长
x = 0:dx:L; % 空间网格
T = zeros(1, N+1); % 温度分布
T(1) = 100; % 边界条件
T(end) = 0; % 边界条件

% 迭代求解
for iter = 1:1000
    for i = 2:N
        T(i) = T(i) + dx^2 * (T(i-1) - 2*T(i) + T(i+1));
    end
end

plot(x, T);

import numpy as np
import matplotlib.pyplot as plt

# 参数设置
L = 1.0  # 区间长度
N = 100  # 网格点数
dx = L / N  # 网格步长
x = np.linspace(0, L, N+1)  # 空间网格
T = np.zeros(N+1)  # 温度分布
T[0] = 100  # 边界条件
T[-1] = 0  # 边界条件

# 迭代求解
for _ in range(1000):
    for i in range(1, N):
        T[i] = T[i] + dx**2 * (T[i-1] - 2*T[i] + T[i+1])

plt.plot(x, T)
plt.xlabel('Position')
plt.ylabel('Temperature')
plt.title('Finite Difference Method for Heat Conduction')
plt.show()