def keplerian(time, p, k, ecc, omega, t0, vsys): from get_rvN import get_rvn from numpy import zeros_like vel = zeros_like(time) get_rvn(time, p, k, ecc, omega, t0, vsys, vel) return vel
xlabel('K') subplot(153) hist(ecc) xlabel('ecc') subplot(154) hist(w) xlabel('w') subplot(155) hist(t0) xlabel('t0') show() ##################################################################### with open('chains/nest-stats.dat', 'r') as f: par_file = f.readlines() npar = int(par_file[-1].strip()[0]) print '# of parameters: ', npar pars = [float(par_file[-i].split()[1]) for i in range(1,npar+1)] pars.reverse() t, v, err = loadtxt('14her.rv', unpack=True) vel = zeros_like(t) get_rvn(t, pars[0], pars[1], pars[2], pars[3], pars[4], pars[5], vel) figure() errorbar(t, v, yerr=err, fmt='o') plot(t, vel, 'r-') show()
from get_rv1 import get_rv # 1 planet from get_rvN import get_rvn # n planets from numpy import * times = linspace(2449460, 2452860, 100) vel1 = zeros_like(times) vel2 = zeros_like(times) get_rv(times, 1425, 10, 0.9, 0.2, 2452000, vel1) get_rvn(times, [1425, 13], [10, 3], [0.9, 0.02], [0.2, 0.3], [2452000, 2451000], 0., vel2) savetxt('test/out_test.txt', zip(times, vel1, vel2))