In [17]:
import rebound
import numpy as np
sim = rebound.Simulation()
sim.G = 4*np.pi**2
sim.add(m=1.)
sim.add(m=1.e-3, a=1., e=0.05, pomega=np.pi/4, inc=np.pi/6, Omega=np.pi, f=np.pi/8, hash="p1")
sim.add(m=1.e-3, a=2., hash="p2")
In [18]:
%matplotlib inline
import matplotlib.pyplot as plt
fig = rebound.OrbitPlot(sim)
In [19]:
Nout = 1000
times = np.linspace(0, 1e4, Nout)
e = np.zeros(Nout)
ps = sim.particles
for i, time in enumerate(times):
sim.integrate(time)
e[i] = sim.particles[1].e
In [20]:
fig, ax = plt.subplots()
ax.plot(times, e)
Out[20]:
In [21]:
sim.integrate(0.)
In [22]:
with open('file.txt', 'w') as f:
for val in e:
f.write('{0}\n'.format(val))
In [ ]: