import matplotlib.pyplot as plt
h, v0, g = 3000, 200, 9.8 #h初始高度,v0初速度,g重力加速度
t, n=0, 30
tmax=(2*h/g)**0.5 #时间的最大值,超过无意义
delta = tmax/(n-1) #每个坐标之间的间隔
xt,yt=[ ],[ ] #创建列表:横坐标,纵坐标
while t<=tmax: #循环计算坐标值
a=v0*t
b=h-1/2*g*t**2
xt.append(a)
yt.append(b)
t=t+delta
plt.plot(xt,yt,'r-') #将绘制的点连成线
plt.grid('on')
plt.axis([0, 5000, 0, h])
plt.show()