Метод Эйлера

import numpy as np
import matplotlib.pyplot as plt
 
plt.style.use('seaborn-poster')
 
print('Task 3')
# Define parameters
betan = np.array([1.0, 1.25, 1.5, 1.75, 2.0])
for beta in betan:
    f = lambda x, y: 1 + 0.4 * np.sin(x) - betan * y  # ODE
    h = 0.1  # Step size
    x = np.arange(0, 1, h)  # Numerical grid
    y0 = 0  # Initial Condition
 
    # Explicit Euler Method
    y = np.zeros(len(x))
    y[0] = y0
 
    for i in range(0, len(x) - 1):
        y[i + 1] = y[i] + h * f(x[i], y[i])
 
plt.figure(figsize=(12, 8))
plt.plot(x, y, 'bo--', label='Approximate')
plt.plot(x, 1 + 0.4 * np.sin(x) - betan * y, 'g', label='Exact')
plt.title('Approximate and Exact Solution \
for Simple ODE')
plt.xlabel('t')
plt.ylabel('f(t)')
plt.grid()
plt.legend(loc='lower right')
plt.show()

line 20, in <module>
    y[i + 1] = y[i] + h * f(x[i], y[i])
ValueError: setting an array element with a sequence.

import numpy as np
import matplotlib.pyplot as plt
 
plt.style.use('seaborn-poster')
plt.figure(figsize=(12, 8))
print('Task 3')
# Define parameters
betan = np.array([1.0, 1.25, 1.5, 1.75, 2.0])
for beta in betan:
    f = lambda x, y: 1 + 0.4 * np.sin(x) - beta * y  # ODE
    h = 0.1  # Step size
    x = np.arange(0, 1, h)  # Numerical grid
    y0 = 0  # Initial Condition
 
    # Explicit Euler Method
    y = np.zeros(len(x))
    y[0] = y0
 
    for i in range(0, len(x) - 1):
        y[i + 1] = y[i] + h * f(x[i], y[i])
    plt.plot(x, y, 'bo--', label=beta)
 
 
plt.title('Approximate and Exact Solution \
for Simple ODE')
plt.xlabel('t')
plt.ylabel('f(t)')
plt.grid()
plt.legend(loc='lower right')
plt.show()