########## Begin ##########
import numpy as np
import matplotlib.pyplot as plt
def calBombTrace(h, v0):
    g,n = 9.8,30
    tmax = (2*h/g)**0.5
    t = np.linspace(0, tmax, n)
    xt = v0*t
    yt = h-1/2*g*t**2
    return xt, yt
H = [3000, 2000, 1000]
V0 = [200, 260, 230]
for h in H:
    for v0 in V0:
        xt,yt = calBombTrace(h,v0)
        plt.plot(xt,yt,'r-')
plt.grid('on')
plt.axis([0, 6500, 0, 3000])
plt.show()

########## End ##########
plt.savefig( 'src/step6/student/pic.png' )
plt.close()