用Python3绘制Sage中欧拉法的图形

import numpy as np
import matplotlib.pyplot as plt

def euler_method(f, x0, y0, h, n):
    x = np.linspace(x0, x0 + n * h, n+1)
    y = np.zeros(n+1)
    y[0] = y0

    for i in range(n):
        y[i+1] = y[i] + h * f(x[i], y[i])

    return x, y

def f(x, y):
    return y  # 示例中的微分方程

x0 = 0  # 初始点 x0
y0 = 1  # 初始点 y0
h = 0.1  # 步长
n = 10  # 步数

x, y = euler_method(f, x0, y0, h, n)

plt.plot(x, y, marker='o', linestyle='-', color='b')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Euler Method')
plt.grid(True)
plt.show()