newton.py

def newton(theta, gamma, thrust, lift, drag, P, m):
    cos_theta = cos(theta)
    sin_theta = sin(theta)
    lift_drag_thrust = np.array([lift, drag, thrust])
    # Z-axis
    F_z = np.sum(lift_drag_thrust * np.array([cos_theta, -sin(gamma), sin_theta])) - P
    # X-axis
    F_x = np.sum(lift_drag_thrust * np.array([-sin_theta, -abs(cos(gamma)), cos_theta]))
    # Compute Acceleration using a = F/m
    return np.array([F_x / m, F_z / m])