#!/home/misha/venv/bin/python
import math as  m
import numpy as np
import matplotlib.pyplot as plt
import os
from pathlib import Path

os.chdir(Path(__file__).parent)
a=np.loadtxt("dim2.dat")

figure,(p1,p2)=plt.subplots(1,2)

v=np.sqrt(a[:,3]**2+a[:,4]**2)
p1.plot(a[:,1],a[:,2],color="red")
p1.legend("trajektoor y(x)")
p2.plot(a[:,0],v[:],color="black")
p1.grid(color="black")
p2.grid(color="black")
plt.show()