How can I setup the three body problem in python? How to I define the function to solve the ODEs?
The three equations are
x'' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * x,
y'' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * y, and
z'' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * z.
Written as 6 first order we have
x' = x2,
y' = y2,
z' = z2,
x2' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * x,
y2' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * y, and
z2' = -mu / np.sqrt(x ** 2 + y ** 2 + z ** 2) * z
I also want to add in the path Plot o Earth's orbit and Mars which we can assume to be circular.
Earth is 149.6 * 10 ** 6 km from the sun and Mars 227.9 * 10 ** 6 km.