(Py/Cy)thon Nbody code