※ 绘制炸弹轨迹 V——绘制多条轨迹

import numpy as np
import matplotlib.pyplot as plt
########## Begin ##########
def calBombTrace(h, v0):
    g = 9.8
    tmax=(2*h/g)**0.5
    t=np.linspace(0,tmax,30)
    xt=v0*t
    yt=h-0.5*g*t**2
    return xt, yt
########## End ##########
xt, yt = calBombTrace(3000, 200) #计算第1种情况时n个点的坐标
plt.plot(xt,yt,'r-')             #绘制第1种情况对应的轨迹
xt, yt = calBombTrace(3000, 260) #计算第2种情况时n个点的坐标
plt.plot(xt,yt,'r-')             #绘制第2种情况对应的轨迹
xt, yt = calBombTrace(2000, 200) #计算第3种情况时n个点的坐标
plt.plot(xt,yt,'r-')             #绘制第3种情况对应的轨迹
xt, yt = calBombTrace(2000, 260) #计算第4种情况时n个点的坐标
plt.plot(xt,yt,'r-')             #绘制第4种情况对应的轨迹
plt.grid('on')
plt.axis([0, 6500, 0, 3000])
plt.show()
plt.savefig( 'src/step5/student/pic.png' )
plt.close()