numpy数组增量时间步的数值积分

import numpy as np

# 假设我们有一个函数f(t, y)，它描述了y随时间t的变化率
def f(t, y):
    return -y  # 例如，简单的指数衰减

# 初始条件
y0 = 1.0  # 初始值
t0 = 0.0  # 初始时间
tf = 1.0  # 最终时间
dt = 0.01  # 时间步长

# 计算需要的时间步数
n_steps = int((tf - t0) / dt)

# 初始化时间和解的数组
t_values = np.linspace(t0, tf, n_steps + 1)
y_values = np.zeros(n_steps + 1)
y_values[0] = y0

# 使用欧拉法进行数值积分
for i in range(n_steps):
    y_values[i + 1] = y_values[i] + f(t_values[i], y_values[i]) * dt

# 打印结果
for t, y in zip(t_values, y_values):
    print(f"t = {t:.2f}, y = {y:.4f}")