diff --git a/orbitize/hipparcos.py b/orbitize/hipparcos.py index 187ed395..6c6a4a15 100644 --- a/orbitize/hipparcos.py +++ b/orbitize/hipparcos.py @@ -521,11 +521,11 @@ def log_prob(model_pars): myHipLogProb.plx0 + 3 * myHipLogProb.plx0_err, 1000, ) - axes[0].hist(sampler.flatchain[:, 0], bins=50, density=True, color="r") + axes[0].hist(sampler.flatchain[:, 0], bins=50, density=True, color="grey") axes[0].plot( xs, norm(myHipLogProb.plx0, myHipLogProb.plx0_err).pdf(xs), color="k" ) - axes[0].set_xlabel("plx [mas]") + axes[0].set_xlabel("$\pi$ [mas]") # PM RA xs = np.linspace( @@ -533,11 +533,11 @@ def log_prob(model_pars): myHipLogProb.pm_ra0 + 3 * myHipLogProb.pm_ra0_err, 1000, ) - axes[1].hist(sampler.flatchain[:, 1], bins=50, density=True, color="r") + axes[1].hist(sampler.flatchain[:, 1], bins=50, density=True, color="grey") axes[1].plot( xs, norm(myHipLogProb.pm_ra0, myHipLogProb.pm_ra0_err).pdf(xs), color="k" ) - axes[1].set_xlabel("PM RA [mas/yr]") + axes[1].set_xlabel("$\mu_{{\\alpha_0^*}}$ [mas/yr]") # PM Dec xs = np.linspace( @@ -545,27 +545,30 @@ def log_prob(model_pars): myHipLogProb.pm_dec0 + 3 * myHipLogProb.pm_dec0_err, 1000, ) - axes[2].hist(sampler.flatchain[:, 2], bins=50, density=True, color="r") + axes[2].hist(sampler.flatchain[:, 2], bins=50, density=True, color="grey") axes[2].plot( xs, norm(myHipLogProb.pm_dec0, myHipLogProb.pm_dec0_err).pdf(xs), color="k" ) - axes[2].set_xlabel("PM Dec [mas/yr]") + axes[2].set_xlabel("$\mu_{{\\delta_0}}$ [mas/yr]") # RA offset - axes[3].hist(sampler.flatchain[:, 3], bins=50, density=True, color="r") + axes[3].hist(sampler.flatchain[:, 3], bins=50, density=True, color="grey") xs = np.linspace( -3 * myHipLogProb.alpha0_err, 3 * myHipLogProb.alpha0_err, 1000 ) axes[3].plot(xs, norm(0, myHipLogProb.alpha0_err).pdf(xs), color="k") - axes[3].set_xlabel("RA Offset [mas]") + axes[3].set_xlabel("$\\alpha_0^*$ [mas]") # Dec offset xs = np.linspace( -3 * myHipLogProb.delta0_err, 3 * myHipLogProb.delta0_err, 1000 ) - axes[4].hist(sampler.flatchain[:, 4], bins=50, density=True, color="r") + axes[4].hist(sampler.flatchain[:, 4], bins=50, density=True, color="grey") axes[4].plot(xs, norm(0, myHipLogProb.delta0_err).pdf(xs), color="k") - axes[4].set_xlabel("Dec Offset [mas]") + axes[4].set_xlabel("$\delta_0$ [mas]") + + for a in axes: + a.set_ylabel('relative prob.') plt.tight_layout() plt.savefig(saveplot, dpi=250) diff --git a/orbitize/plot.py b/orbitize/plot.py index 0b5a1ef3..9eb85a3b 100644 --- a/orbitize/plot.py +++ b/orbitize/plot.py @@ -74,24 +74,24 @@ def plot_corner(results, param_list=None, **corner_kwargs): default_labels = { "sma": "$a_{{B}}$ [au]", - "ecc": "$ecc_{{B}}$", - "inc": "$inc_{{B}}$ [$^\\circ$]", + "ecc": "$ecc$", + "inc": "$i$ [$^\\circ$]", "aop": "$\\omega_{{B}}$ [$^\\circ$]", - "pan": "$\\Omega_{{B}}$ [$^\\circ$]", - "tau": "$\\tau_{{B}}$", + "pan": "$\\Omega$ [$^\\circ$]", + "tau": "$\\tau$", "tp": "$T_{{\\mathrm{{P}}}}$", "plx": "$\\pi$ [mas]", "gam": "$\\gamma$ [km/s]", "sig": "$\\sigma$ [km/s]", "mtot": "$M_T$ [M$_{{\\odot}}$]", - "m0": "$M_{{A}}$ [M$_{{\\odot}}$]", - "m": "$M_{{B}}$ [M$_{{\\odot}}$]", + "m0": "$M_{{a}}$ [M$_{{\\odot}}$]", + "m": "$M_{{b}}$ [M$_{{\\odot}}$]", "pm_ra": "$\\mu_{{\\alpha^{{*}}}}$ [mas/yr]", "pm_dec": "$\\mu_{{\\delta}}$ [mas/yr]", "alpha0": "$\\alpha^{{*}}_{{0}}$ [mas]", "delta0": "$\\delta_0$ [mas]", # "m": "$M_{0}$ [M$_{{\\rm Jup}}$]", - "per": "$P_{{B}}$ [yr]", + "per": "$P$ [yr]", "K": "$K_{0}$ [km/s]", "x": "$X_{0}$ [AU]", "y": "$Y_{0}$ [AU]",