From 5861175f409b4cbee3bd39b8af9a47491af6bda5 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Sat, 22 Oct 2016 19:24:13 +0200
Subject: [PATCH] Fix plot routines when ndim=1
---
pyfstat.py | 20 ++++++++++++++++++--
1 file changed, 18 insertions(+), 2 deletions(-)
diff --git a/pyfstat.py b/pyfstat.py
index e132c52..d271ab8 100755
--- a/pyfstat.py
+++ b/pyfstat.py
@@ -785,6 +785,16 @@ class MCMCSearch(BaseSearchClass):
add_prior=False, nstds=None, label_offset=0.4,
dpi=300, rc_context={}, **kwargs):
+ if self.ndim < 2:
+ with plt.rc_context(rc_context):
+ fig, ax = plt.subplots(figsize=figsize)
+ ax.hist(self.samples, bins=50, histtype='stepfilled')
+ ax.set_xlabel(self.theta_symbols[0])
+
+ fig.savefig('{}/{}_corner.png'.format(
+ self.outdir, self.label), dpi=dpi)
+ return
+
with plt.rc_context(rc_context):
fig, axes = plt.subplots(self.ndim, self.ndim,
figsize=figsize)
@@ -1008,9 +1018,15 @@ class MCMCSearch(BaseSearchClass):
if symbols:
axes[i].set_ylabel(symbols[i])
else:
+ axes[0].ticklabel_format(useOffset=False, axis='y')
cs = chain[:, :, temp].T
- axes.plot(cs, color='k', alpha=alpha)
- axes.ticklabel_format(useOffset=False, axis='y')
+ if burnin_idx:
+ axes[0].plot(idxs[:burnin_idx], cs[:burnin_idx],
+ color="r", alpha=alpha, lw=lw)
+ axes[0].plot(idxs[burnin_idx:], cs[burnin_idx:], color="k",
+ alpha=alpha, lw=lw)
+ if symbols:
+ axes[0].set_ylabel(symbols[0])
axes.append(fig.add_subplot(ndim+1, 1, ndim+1))
lnl = sampler.lnlikelihood[temp, :, :]
--
GitLab