diff options
Diffstat (limited to 'utils/dm-search')
-rwxr-xr-x | utils/dm-search | 7 |
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)]) + '$') |