update sbc plot to include grey band

1 files changed, 5 insertions, 2 deletions

diff --git a/utils/chi2 b/utils/chi2
index 6764dd5..4c0eddd 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -648,8 +648,11 @@ if __name__ == '__main__':
         fig = plt.figure()
         axes = []
         for i, name in enumerate(FIT_PARS):
-            axes.append(plt.subplot(3,3,i+1))
-            plt.hist(pull[i],bins=np.linspace(0,100,101),histtype='step',normed=True)
+            axes.append(plt.subplot(3,2,i+1))
+            n, bins, patches = plt.hist(pull[i],bins=np.linspace(0,100,11),histtype='step')
+            expected = len(pull[i])/(len(bins)-1)
+            plt.axhline(expected,color='k',ls='--',alpha=0.25)
+            plt.axhspan(poisson.ppf(0.005,expected), poisson.ppf(0.995,expected), facecolor='0.5', alpha=0.25)
             plt.title(name)
         for ax in axes:
             despine(ax=ax,left=True,trim=True)