plot_HS_mass.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

plot_HS_mass.py
View Options

	import matplotlib.pyplot as plt
	import pylab as P
	import numpy as np

	P.rc('text', usetex=True)
	P.rc('text.latex', preamble='\usepackage{amsmath}\usepackage{color}')
	font = {'size' : 16}
	P.rc('font', **font)
	P.rc('grid', linewidth=1,color='#666666')

	files = ["HS_mass_pdf1.dat",\
	"HS_mass_pdf2.dat",\
	"HS_mass_pdf3.dat"]

	ls = [ ['blue','dotted'], ['red','solid'], ['green','dashed']]
	labels = ['box', 'Gaussian', 'box+Gaussian']

	fig, ax = plt.subplots()

	for i,f in enumerate(files):
	data = np.loadtxt(f)
	dataT = zip(*data)

	chi2min = min(dataT[4])
	print chi2min
	# minindex = dataT[5].index(chi2min)

	ax.plot(dataT[0],dataT[4],color=ls[i][0],linestyle=ls[i][1], linewidth=2, label=labels[i])

	ax.set_xlim([min(dataT[0]),max(dataT[0])])
	# ax.set_ylim([0.,200.])

	if dataT[7][0] == 0:
	ax.text(0.02,1.015,r"$\Delta m_H^\text{theo} = "+format(dataT[1][0],'4.2f')+"~\mathrm{GeV}$, $\Lambda = \mathrm{default}$",transform=ax.transAxes)
	else:
	ax.text(0.02,1.015,r"$\Delta m_H^\text{theo} = "+format(dataT[1][0],'4.2f')+"~\mathrm{GeV}$, $\Lambda = "+format(dataT[7][0],'3.1f')+"$",transform=ax.transAxes)
	# ax.text(0.42,1.02,r",transform=ax.transAxes)
	plt.grid()

	leg = plt.legend(fancybox = True, fontsize=14, loc=3)
	leg.get_frame().set_alpha(0.5)

	plt.xlabel(r'$m_{H}~[\mathrm{GeV}]$')
	plt.ylabel(r'$\chi^2$')

	plt.savefig('HS_mass.pdf')