import numpy as np
from scipy.integrate import solve_ivp

G = 9.81

def odeFunc(t,y,a):

    # y = [x,v], y_dot = [v,-g]

    # y = [x,y,z,u,v,w,ax,ay,az]
    return [y[1],-G+a]


sol = solve_ivp(odeFunc,[0,4],[100,0],t_eval=[4],args=[9.81])

print(sol.y[0][0])

