import matplotlib.pyplot as plt
from numpy import linspace, full_like, array, arange
from sympy import *
print("------ User input ------------------------")
A = 6
B = 3
AB = A/B
AB2 = AB*AB
v = var('v')
print -AB
print AB2
print v*v/B
x = - AB + sqrt(AB2 + v*v/B)
via = linspace(float(0), float(6))
x = lambdify(v,x,"numpy")
x = array(x(via))
tmp = plt.plot(via, x, label="$x / \mathrm{m}$")
tmp = plt.xlabel("$v / \mathrm{m / s}$")
tmp = plt.legend(loc="best")
plt.savefig("3.1.B_1.svg", transparent=True)
plt.show()