# -*- coding: utf-8 -*-
"""
Created on Sun Oct  9 11:00:22 2022

@author: Chris
Lab 10 Q1
"""

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    f = x*(1+4*y**2)
    return f

x0 = 0
y0 = 0
h = 0.1

xmax = 1.0

xvals = np.array([x0])
yvals = np.array([y0])

x = x0
y = y0
while x<xmax:
    yn = y + h*f(x,y)
    xn = x + h

    xvals = np.append(xvals,xn)
    yvals = np.append(yvals,yn)
    x,y = xn,yn
    #print(xn,yn)
    
plt.plot(xvals,yvals)