initial_velocity = 26.85
from math import cos, sin,pi
import matplotlib.pyplot as plt
def horiz(t,angle):
    return t * initial_velocity * cos(angle / 180.0 * pi)
def vert(t,angle):
    return t * initial_velocity * sin(angle / 180.0 * pi) - 4.9 * t**2

angles = [30,35,40,45,50,55,60]
hs = []
vs = []
for angle in angles:
    t = range(int(25*sin(angle/180.0*pi)))
    for i in range(len(t)):
        t[i] = t[i] / 4.0
    t.append(t[-1] + 0.25)
    t.append(25*sin(angle/180.0*pi) / 4)
    h = []
    v = []
    for time in t:
        h.append(horiz(time,angle))
        v.append(vert(time,angle))
    hs.append(h)
    vs.append(v)
plt.plot(hs[0],vs[0],'r',label="30 degrees")
plt.plot(hs[1],vs[1],'b',label ="35 degrees")
plt.plot(hs[2],vs[2],'g',label = "40 degrees")
plt.plot(hs[3],vs[3],'c',label = "45 degrees")
plt.plot(hs[4],vs[4],'m',label = "50 degrees")
plt.plot(hs[5],vs[5],'y',label = "55 degrees")
plt.plot(hs[6],vs[6],'k',label = "60 degrees")
plt.legend()
plt.ylim(0,30)
plt.xlim(0,75)
plt.text(0,34,"Made by Mike Ambrose")
plt.ylabel("Vertical Displacement (m)")
plt.xlabel("Horizontal Displacement (m)")
plt.title("Vertical Displacement vs Horizontal Displacement")
plt.show()