import matplotlib.pyplot as plt
import numpy as np

from math import sqrt


H=20
m=1
VV=5
S=0.5
C=0.4
ro=1.3
Vx=0
Vy=0
y=H
x=0
dt=0.1
XX=[0]
YY=[H]
while y>=0:
    V0=sqrt(Vx*Vx+Vy*Vy)
    ay=9.8-(C*ro*S*V0*Vy/2)/m
    Vy=Vy+ay*dt
    y=y-Vy*dt-ay*dt*dt/2
    YY.append(y)
    ax=C*ro*S*V0*(VV-Vx)/2/m
    Vx=Vx+ax*dt
    x=x+Vx*dt+ax*dt*dt/2
    XX.append(x)
plt.plot(XX,YY,'ro')
plt.show()