возникла проблема по построению графика баллистического движения на Python
Пытаюсь сделать демонстрацию графика через matplotlib, но не получается задать правильно координату Y. Понимаю, что нужно найти максимальную высоту по формуле, проделать координаты через arange до неё, а потом добавить в список координаты от этой максимальной высоты, до 0. Ну и в итоге сделать вывод графика. вот код:
import numpy as np
import matplotlib.pyplot as plt
# Compute the x and y coordinates for points on a sine curve
X0 = int(input("Введите X начальное")) #0
Y0 = int(input("Введите Y начальное")) #0
G = int(input("Введите g")) # 10
a = float(input("Введите угол запуска снаряда")) # 45
t = int(input("Введите t ")) # 5
V0 = int(input("Введите V начальное ")) # 15
formul_1 = V0 * np.cos(a)*t
formul_2 = V0*np.sin(a)*t-G*(t**2)/2
H = ((V0**2)*(np.sin(a)**2))/(2*G)
x = np.arange(0, formul_1, 1)
y = x
print(x, "\n", y)
print("Plot the points using matplotlib:")
plt.plot(x, y)
plt.show()
Надеюсь, кто-нибудь сможет помочь с координатой Y.
Источник: Stack Overflow на русском