aboutsummaryrefslogtreecommitdiff
path: root/utils/dm-search
diff options
context:
space:
mode:
Diffstat (limited to 'utils/dm-search')
-rwxr-xr-xutils/dm-search7
1 files changed, 6 insertions, 1 deletions
diff --git a/utils/dm-search b/utils/dm-search
index 872004b..1e9c987 100755
--- a/utils/dm-search
+++ b/utils/dm-search
@@ -27,7 +27,6 @@ from scipy.stats import iqr, poisson
from matplotlib.lines import Line2D
from scipy.stats import iqr, norm, beta, percentileofscore
from scipy.special import spence
-from itertools import izip_longest
from sddm.stats import *
from sddm.dc import estimate_errors, EPSILON, truncnorm_scaled
import emcee
@@ -37,6 +36,7 @@ from scipy.integrate import quad
from sddm.dm import *
from sddm import SNOMAN_MASS, AV_RADIUS
import nlopt
+from itertools import chain
# Likelihood Fit Parameters
# 0 - Atmospheric Neutrino Flux Scale
@@ -145,6 +145,11 @@ def plot_hist2(hists, bins, color=None):
bincenters = (bins[1:] + bins[:-1])/2
plt.hist(bincenters, bins=bins, histtype='step', weights=hists[id],color=color)
plt.gca().set_xscale("log")
+ major = np.array([10,100,1000,10000])
+ minor = np.unique(list(chain(*list(range(i,i*10,i) for i in major[:-1]))))
+ minor = np.setdiff1d(minor,major)
+ plt.gca().set_xticks(major)
+ plt.gca().set_xticks(minor,minor=True)
plt.xlabel("Energy (MeV)")
plt.title('$' + ''.join([particle_id[int(''.join(x))] for x in grouper(str(id),2)]) + '$')