import numpy as np
import matplotlib.pyplot as plt

def y_komp(v0y, t):
     g = 9.81 
     return v0y*t - 0.5*g*t**2

   
def x_komp(v0x, t):
    return v0x*t


tid = np.linspace(0, 22, 100)

v0 = 150
vinkel = np.pi/4

vx_init = v0*np.cos(vinkel)
vy_init = v0*np.sin(vinkel)

Xkomp = x_komp(vx_init, tid)
Ykomp = y_komp(vy_init, tid)

plt.close('all')
plt.plot(Xkomp, Ykomp)
plt.xlabel('x [m]')
plt.ylabel('y [m]')
plt.grid()

# plt.savefig('plott_kanonkule.pdf')
plt.show()