python欧拉法

2021/10/7 11:11:04

本文主要是介绍python欧拉法,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

import numpy as np
import matplotlib.pyplot as plt

# Problem1
h = [0.001, 0.005, 0.01, 0.05, 0.1]
num = 5 / h[0]
y = 1
t = 0
dy_dt = -1
tPlot = np.arange(0, 5, h[0])
yPlot = []
for i in range(int(num)):
    yPlot.append(y)
    t += h[0]
    y += dy_dt * h[0]
    dy_dt = -2*y+2-np.exp(-4*t)
    if i % 1000 == 999:
        print(t, y)
plt.plot(tPlot, yPlot, label="Euler's method")
yExact = (np.exp(-4*tPlot))/2-(np.exp(-2*tPlot))/2+1
plt.plot(tPlot, yExact, label="exact solution")
plt.xlabel("t")
plt.ylabel("y")
plt.ylim(0.8, 1.05)
plt.legend()
plt.show()

plt.plot(tPlot, yPlot, label="0.001")
for i in range(4):
    num = 5 / h[i+1]
    y = 1
    t = 0
    dy_dt = -1
    tPlot = np.arange(0, 5, h[i+1])
    yPlot = []
    for j in range(int(num)):
        yPlot.append(y)
        t += h[i+1]
        y += dy_dt * h[i+1]
        dy_dt = -2*y+2-np.exp(-4*t)
    plt.plot(tPlot, yPlot, label=h[i+1])
plt.xlabel("t")
plt.ylabel("y")
plt.ylim(0.8, 1.05)
plt.legend()
plt.show()


这篇关于python欧拉法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程