import numpy as np
import matplotlib.pyplot as plt
plt.style.use('ff.mplstyle')

x = [0.2, 0.4, 0.6, 0.8, 1.0]                  # м
v2_1 = np.array([9, 18, 26, 29, 42]) * 1e-3    # м^2/с^2
v2_2 = np.array([15, 34, 47, 65, 85]) * 1e-3   # м^2/с^2
v2_3 = np.array([20, 49, 75, 95, 127]) * 1e-3  # м^2/с^2

plt.scatter(x, v2_1, label=r'$\alpha = 0.11^\circ$')
plt.scatter(x, v2_2, label=r'$\alpha = 0.23^\circ$')
plt.scatter(x, v2_3, label=r'$\alpha = 0.37^\circ$')
plt.xlim(xmin=0)
plt.ylim(ymin=1e-6)
plt.xlabel('$x$, м')
plt.ylabel(r'$v^2, \frac{м^2}{c^2}$')
plt.title('Зависимость квадрата скорости тележки $v^2$ \n от пройденного ей расстояния $x$. Случай тележки без груза')
plt.legend(title='Наклон скамьи')
plt.savefig('example_3_sameaxis.pdf')