Function euler

ode.py:6–44

Calculate the solution of the initial-value problem (IVP). Solve the IVP from the Euler method. Args: f (function): equation f(x). a (float): the initial point. b (float): the final point. n (int): number of intervals. ya (numpy.ndarray): initial val

(f, a, b, n, ya)

def euler(f, a, b, n, ya):
    """Calculate the solution of the initial-value problem (IVP).

    Solve the IVP from the Euler method.

    Args:
        f (function): equation f(x).
        a (float): the initial point.
        b (float): the final point.
        n (int): number of intervals.
        ya (numpy.ndarray): initial values.

    Returns:
        vx (numpy.ndarray): x values.
        vy (numpy.ndarray): y values (solution of IVP).
    """
    vx = np.zeros(n)
    vy = np.zeros(n)

    h = (b - a) / n
    x = a
    y = ya

    vx[0] = x
    vy[0] = y

    fxy = f(x, y)
    print(f"i = 000,\tx = {x:+.4f},\ty = {y:+.4f}")

    for i in range(0, n):
        x = a + (i + 1) * h
        y += h * fxy

        fxy = f(x, y)
        print(f"i = {(i+1):03d},\tx = {x:+.4f},\ty = {y:+.4f}")
        vx[i] = x
        vy[i] = y

    return vx, vy


def taylor2(f, df1, a, b, n, ya):