def main(): filenames = sys.argv[1:] print("Will open files:") for f in filenames: print(f) data = tbt.chain_pickle_files(filenames) mupt = data['mupt'] muisPF = data['muisPF'] muist = data['muist'] muiso04 = data['muiso04'] ################################################################################ plt.figure() plt.subplot(3, 3, 1) lch.hist_err(mupt, bins=100, range=(0, 100), color='k') plt.subplot(3, 3, 2) lch.hist_err(muiso04[muiso04 > 0], bins=100, range=(0, .20), color='k') plt.subplot(3, 3, 5) lch.hist_2D(mupt[muiso04 > 0], muiso04[muiso04 > 0], xbins=100, ybins=100, xrange=(0, 100), yrange=(0, .20)) #plt.xlim(50,150) #plt.ylim(0, 3.2) ################################################################################ # Cut on the wmass #index = wmass>70.0 #index *= wmass<95.0 plt.show()
#lch.hist_err(btags[jetpt>=30],bins=100,range=(0,1.25)) plt.hist(btags[jetpt >= 30], bins=100, range=(0, 1.25)) plt.xlabel("CSV2 b-tag output, jet p$_T >= $ 30", fontsize=18) plt.savefig("/home/me25hage/Top_BNV/test/images/btagplus30.png") plt.figure() #lch.hist_err(btags, range = [0,1.25],bins=100,markersize=5) plt.hist(btags, range=[0, 1.1], bins=110) plt.xlabel("CSV2 b-tag output", fontsize=18) plt.tight_layout() plt.savefig('/home/me25hage/Top_BNV/test/images/btag.png') plt.figure() plt.plot(btags[(btags >= 0) * (jetpt < 100)], jetpt[(btags >= 0) * (jetpt < 100)], '.', alpha=0.1) #plt.savefig('home/../images/btag.png') plt.figure() #plt.plot(btags[(btags>=0)*(jetpt<100)],jetpt[(btags>=0)*(jetpt<100)],'.',alpha=0.1) lch.hist_2D(btags[(btags >= 0) * (jetpt < 100)], jetpt[(btags >= 0) * (jetpt < 100)], xbins=100, ybins=100, xrange=(0, 1), yrange=(0, 100)) plt.xlabel(r"b-tagging variable", fontsize=18) plt.show()
print "Events: ", len(mass0[mass0 > 0]) plt.figure() lch.hist_err(mass0[mass0 > 0], bins=125, range=(0, 1000)) plt.xlabel("mass0") # plot of highest pt plt.figure() lch.hist_err(pt0_max[pt0_max > 0], bins=125, range=(0, 1000)) plt.xlabel("pt0_max") plt.figure() lch.hist_err(mass1[mass1 > 0], bins=125, range=(0, 1000)) plt.xlabel("mass1") plt.figure() lch.hist_2D(mass0, mass1, xbins=100, ybins=100, xrange=(0, 500), yrange=(0, 500)) plt.xlabel("mass0") plt.ylabel("mass1") plt.figure() lch.hist_2D(mass0, pt0_max, xbins=100, ybins=100, xrange=(0, 500), yrange=(0, 500)) plt.xlabel("mass0") plt.ylabel("pt0_max") ####################### # plot all of the masses in column one plt.figure() lch.hist_err(unique_m0[unique_m0 > 0], bins=125, range=(0, 1000)) plt.xlabel("unique_m0") #######################
def main(): #lumi_file_name = 'lumi_info.pkl' #lumi_info = pickle.load( open( lumi_file_name, "rb" ) ) filenames = sys.argv[1:] #data,tot_lumi = tbt.chain_pickle_files(filenames,lumi_info) #print("tot_lumi: ",tot_lumi) T = ROOT.TChain("T") print("Will open files:") for f in filenames: print(f) T.Add(f) ntops = [] topmass = [] wmass = [] csvs = [] angles = [] dRs = [] wH = [] njets = [] leadmupt = [] leadmueta = [] subleadmupt = [] subleadmueta = [] metpt = [] triggers = [[], [], [], []] nentries = T.GetEntries() for i in range(nentries): T.GetEntry(i) njets.append(T.njet) ntop = T.ntop ntops.append(ntop) for n in range(ntop): topmass.append(T.topmass[n]) wmass.append(T.wmass[n]) dRs.append(T.wdR[n]) wH.append(T.wH[n]) angles.append(T.wangle[n]) for n in range(T.njet): csvs.append(T.jetcsv[n]) leadmupt.append(T.leadmupt) leadmueta.append(T.leadmueta) subleadmupt.append(T.subleadmupt) subleadmueta.append(T.subleadmueta) metpt.append(T.METpt) triggers[0].append(T.trig_HLT_IsoMu24_accept) triggers[1].append(T.trig_HLT_IsoTkMu24_accept) triggers[2].append(T.trig_HLT_IsoMu22_eta2p1_accept) triggers[3].append(T.trig_HLT_IsoTkMu22_eta2p1_accept) ntops = np.array(ntops) topmass = np.array(topmass) wmass = np.array(wmass) csvs = np.array(csvs) angles = np.array(angles) dRs = np.array(dRs) wH = np.array(wH) njets = np.array(njets) leadmupt = np.array(leadmupt) leadmueta = np.array(leadmueta) subleadmupt = np.array(subleadmupt) subleadmueta = np.array(subleadmueta) metpt = np.array(metpt) triggers[0] = np.array(triggers[0]) triggers[1] = np.array(triggers[1]) triggers[2] = np.array(triggers[2]) triggers[3] = np.array(triggers[3]) ''' for a in zip(topmass, wmass, csvs, angles, dRs, njets): print(a) a = np.array(a) ''' ################################################################################ plt.figure() plt.subplot(3, 3, 1) lch.hist_err(topmass, bins=100, range=(0, 600), color='k') plt.xlabel('Top Mass (GeV)') plt.subplot(3, 3, 2) lch.hist_err(wmass, bins=100, range=(0, 300), color='k') plt.xlabel('W Mass (GeV)') plt.subplot(3, 3, 3) lch.hist_err(csvs, bins=110, range=(-11, 1.1), color='k') plt.xlabel('CSV variable') plt.subplot(3, 3, 4) lch.hist_err(angles, bins=100, range=(0, 3.2), color='k') plt.xlabel('Angles') plt.subplot(3, 3, 5) #plt.plot(wmass,angles,'.',markersize=0.5,alpha=0.2) lch.hist_2D(wmass, angles, xbins=100, ybins=100, xrange=(0, 300), yrange=(0, 3.14)) plt.xlim(50, 150) plt.ylim(0, 3.2) plt.xlabel('W Mass') plt.ylabel('Angles') plt.subplot(3, 3, 6) lch.hist_err(dRs, bins=100, range=(0, 3.2), color='k') plt.xlabel('dRs') plt.subplot(3, 3, 7) lch.hist_2D(dRs, angles, xbins=100, ybins=100, xrange=(0, 6.28), yrange=(0, 3.14)) plt.xlabel('dRs') plt.ylabel('Angles') plt.subplot(3, 3, 8) lch.hist_err(wH, bins=100, range=(0, 250), color='k') plt.xlabel('scalar H') plt.subplot(3, 3, 9) lch.hist_err(ntops, bins=6, range=(0, 6), color='k') plt.xlabel('ntops') plt.tight_layout() ################################################################################ # Cut on the wmass index = wmass > 70.0 index *= wmass < 95.0 #index = (np.abs(angles - dRs)<=0.45) plt.figure() plt.title('W Mass Cuts') plt.subplot(3, 3, 1) lch.hist_err(topmass[index], bins=100, range=(0, 600), color='k') plt.xlabel('Top Mass (GeV)') plt.subplot(3, 3, 2) lch.hist_err(wmass[index], bins=100, range=(0, 300), color='k') plt.xlabel('W Mass (GeV)') plt.subplot(3, 3, 4) lch.hist_err(angles[index], bins=100, range=(0, 3.2), color='k') plt.xlabel('Angles') plt.subplot(3, 3, 5) #plt.plot(wmass[index],angles[index],'.',markersize=0.5,alpha=0.2) lch.hist_2D(wmass[index], angles[index], xbins=100, ybins=100, xrange=(0, 300), yrange=(0, 3.14)) plt.xlim(50, 150) plt.ylim(0, 3.2) plt.xlabel('W Mass') plt.ylabel('Angles') plt.subplot(3, 3, 6) lch.hist_err(dRs[index], bins=100, range=(0, 3.2), color='k') plt.xlabel('dRs') plt.subplot(3, 3, 7) lch.hist_2D(dRs[index], angles[index], xbins=100, ybins=100, xrange=(0, 6.28), yrange=(0, 3.14)) plt.xlabel('dRs') plt.ylabel('Angles') plt.subplot(3, 3, 8) lch.hist_err(wH[index], bins=100, range=(0, 250), color='k') plt.xlabel('scalar H') plt.tight_layout() ############################################################################ # Muons ############################################################################ plt.figure() plt.subplot(3, 3, 1) lch.hist_err(leadmupt, bins=100, range=(0, 250), color='k') plt.xlabel(r'Leading muon p$_{T}$ (GeV/c)') plt.subplot(3, 3, 2) lch.hist_err(leadmueta, bins=100, range=(-3.0, 3.0), color='k') plt.xlabel(r'Leading muon $\eta$ (GeV/c)') plt.subplot(3, 3, 4) lch.hist_err(subleadmupt, bins=100, range=(0, 250), color='k') plt.xlabel(r'Sub-leading muon p$_{T}$ (GeV/c)') plt.subplot(3, 3, 5) lch.hist_err(subleadmueta, bins=100, range=(-3.0, 3.0), color='k') plt.xlabel(r'Sub-leading muon $\eta$ (GeV/c)') plt.subplot(3, 3, 7) lch.hist_err(metpt, bins=100, range=(0.0, 100.0), color='k') plt.xlabel(r'Missing E$_T$ (GeV)') plt.tight_layout() ''' # For talk plt.figure() lch.hist_err(topmass[index],bins=100,range=(0,600),color='k') plt.hist(topmass[index],bins=100,range=(0,600),color='grey',alpha=0.2) plt.xlabel(r'Top candidate with W-mass cut (GeV/c$^2$)',fontsize=14) plt.tight_layout() plt.savefig('top.png') plt.figure() lch.hist_err(wmass,bins=100,range=(0,300),color='k') plt.hist(wmass,bins=100,range=(0,300),color='grey',alpha=0.2) plt.xlabel(r'W candidate (GeV/c$^2$)',fontsize=14) plt.tight_layout() plt.savefig('W.png') plt.figure() lch.hist_err(csvs,bins=110,range=(0,1.1),color='k') plt.hist(csvs,bins=100,range=(0,1.1),color='grey',alpha=0.2) plt.xlabel(r'CSVv2 variable',fontsize=14) plt.tight_layout() plt.savefig('csvv2.png') ''' ############################################################################ # Triggers ############################################################################ plt.figure() plt.subplot(5, 4, 1) lch.hist_err(triggers[0], bins=2, range=(0, 2), color='k') plt.xlabel(r'Trigger 0') plt.subplot(5, 4, 2) lch.hist_err(triggers[1], bins=2, range=(0, 2), color='k') plt.xlabel(r'Trigger 1') plt.subplot(5, 4, 3) lch.hist_err(triggers[2], bins=2, range=(0, 2), color='k') plt.xlabel(r'Trigger 2') plt.subplot(5, 4, 4) lch.hist_err(triggers[3], bins=2, range=(0, 2), color='k') plt.xlabel(r'Trigger 3') for i in range(0, 4): for j in range(0, 4): plt.subplot(5, 4, 5 + i * 4 + j) lch.hist_err(triggers[i][triggers[j] == 1], bins=2, range=(0, 2)) plt.tight_layout() plt.show()
def main(): lumi_file_name = 'lumi_info.pkl' lumi_info = pickle.load(open(lumi_file_name, "rb")) filenames = sys.argv[1:] print("Will open files:") for f in filenames: print(f) data, tot_lumi = tbt.chain_pickle_files(filenames, lumi_info) print("tot_lumi: ", tot_lumi) topmass = data['topmass'] wmass = data['wmass'] csvs = data['csvs'] angles = data['angles'] dRs = data['dRs'] #njets = data['njets'] njets = data['njets'] ################################################################################ plt.figure() plt.subplot(3, 3, 1) lch.hist_err(topmass, bins=100, range=(0, 600), color='k') plt.xlabel('Top Mass (GeV)') plt.subplot(3, 3, 2) lch.hist_err(wmass, bins=100, range=(0, 300), color='k') plt.xlabel('W Mass (GeV)') plt.subplot(3, 3, 3) lch.hist_err(csvs, bins=110, range=(0, 1.1), color='k') plt.xlabel('Isolation Variable') plt.subplot(3, 3, 4) lch.hist_err(angles, bins=100, range=(0, 3.2), color='k') plt.xlabel('Angles') plt.subplot(3, 3, 5) #plt.plot(wmass,angles,'.',markersize=0.5,alpha=0.2) lch.hist_2D(wmass, angles, xbins=100, ybins=100, xrange=(0, 300), yrange=(0, 3.14)) plt.xlim(50, 150) plt.ylim(0, 3.2) plt.xlabel('W Mass') plt.ylabel('Angles') plt.subplot(3, 3, 6) lch.hist_err(dRs, bins=100, range=(0, 3.2), color='k') plt.xlabel('dRs') plt.subplot(3, 3, 7) lch.hist_2D(dRs, angles, xbins=100, ybins=100, xrange=(0, 6.28), yrange=(0, 3.14)) plt.xlabel('dRs') plt.ylabel('Angles') ################################################################################ # Cut on the wmass index = wmass > 70.0 index *= wmass < 95.0 plt.figure() plt.title('W Mass Cuts') plt.subplot(3, 3, 1) lch.hist_err(topmass[index], bins=100, range=(0, 600), color='k') plt.xlabel('Top Mass (GeV)') plt.subplot(3, 3, 2) lch.hist_err(wmass[index], bins=100, range=(0, 300), color='k') plt.xlabel('W Mass (GeV)') plt.subplot(3, 3, 4) lch.hist_err(angles[index], bins=100, range=(0, 3.2), color='k') plt.xlabel('Angles') plt.subplot(3, 3, 5) #plt.plot(wmass[index],angles[index],'.',markersize=0.5,alpha=0.2) lch.hist_2D(wmass[index], angles[index], xbins=100, ybins=100, xrange=(0, 300), yrange=(0, 3.14)) plt.xlim(50, 150) plt.ylim(0, 3.2) plt.xlabel('W Mass') plt.ylabel('Angles') plt.subplot(3, 3, 6) lch.hist_err(dRs[index], bins=100, range=(0, 3.2), color='k') plt.xlabel('dRs') plt.subplot(3, 3, 7) lch.hist_2D(dRs[index], angles[index], xbins=100, ybins=100, xrange=(0, 6.28), yrange=(0, 3.14)) plt.xlabel('dRs') plt.ylabel('Angles') ''' # For talk plt.figure() lch.hist_err(topmass[index],bins=100,range=(0,600),color='k') plt.hist(topmass[index],bins=100,range=(0,600),color='grey',alpha=0.2) plt.xlabel(r'Top candidate with W-mass cut (GeV/c$^2$)',fontsize=14) plt.tight_layout() plt.savefig('top.png') plt.figure() lch.hist_err(wmass,bins=100,range=(0,300),color='k') plt.hist(wmass,bins=100,range=(0,300),color='grey',alpha=0.2) plt.xlabel(r'W candidate (GeV/c$^2$)',fontsize=14) plt.tight_layout() plt.savefig('W.png') plt.figure() lch.hist_err(csvs,bins=110,range=(0,1.1),color='k') plt.hist(csvs,bins=100,range=(0,1.1),color='grey',alpha=0.2) plt.xlabel(r'CSVv2 variable',fontsize=14) plt.tight_layout() plt.savefig('csvv2.png') ''' plt.show() return data
print "Making the plots....." plt.figure() <<<<<<< HEAD #lch.hist_2D(theta,top_momentum[top_momentum>-999], xbins=50,ybins=50) #lch.hist_2D(top_mass[top_mass>-999],top_momentum[top_momentum>-999], xbins=50, ybins=50) #lch.hist_2D(theta,second_top_mass,xbins=10,ybins=10) ''' hist,xedges,yedges = np.histogram2d(top_momentum[top_momentum>-999],theta,bins=(25,50)) extent = [xedges[0],xedges[-1],yedges[0],yedges[-1]] plt.imshow(hist,extent=extent,interpolation='nearest',origin='lower') ''' top_momentum = top_momentum[top_momentum>-999] theta = np.array(theta) lch.hist_2D(top_momentum,theta,xrange=(0,50),yrange=(0,3.2),xbins=50,ybins=50) #lch.hist_2D(top_momentum,theta,xbins=50,ybins=50) plt.show() ======= lch.hist_2D(top_momentum[top_momentum>-999],theta[theta>-999],xbins=25,ybins=50,xrange=(0,35),yrange=(0,3.2)) plt.title("%s: Theta vs. Top Momentum" % (tag)) >>>>>>> dafd5f4d19be935ee76309b719ffe9ed78513fdf plt.figure() lch.hist_2D(top_momentum[top_momentum>-999],top_mass[top_momentum>-999],xbins=50,ybins=50,xrange=(0,35),yrange=(0,300)) plt.title("%s: Top Mass vs. Top Momentum" % (tag))
print('elecCut', elecCut) print('muonCut', muonCut) print('nentries', nentries) print('Test', test) print('electrons', electrons) plt.figure() plt.hist(differencesElectron, bins=100) #,range = (0,1)) plt.xlabel("Difference in pt") plt.figure() plt.hist(differencesElectron, bins=100, range=(0, 10)) plt.xlabel("Difference in pt") plt.figure() plt.hist(distance, bins=100, range=(0, .005)) plt.xlabel("Distance between electrons in eta-phi space") plt.figure() lch.hist_2D(distance, differencesElectron) plt.figure() plt.plot(etaElectrons, phiElectrons, 'ro', alpha=0.1, label="Electrons") plt.plot(etaWchildren, phiWchildren, 'b*', alpha=0.1, label="MC Truth") plt.xlabel('eta') plt.ylabel('phi') plt.legend() plt.show()
'''plt.figure(2) plt.subplot(2, 1, 1) #plt.hist(del_R_mu_other_b,bins=50) lkn.hist_err(minimal_dR,bins=100) plt.title("Bottom") plt.xlabel(r"Abs($\delta$R)") plt.subplot(2, 1, 2) #plt.hist(del_R_mu_b,bins=50) lkn.hist_err(minimal_pt,bins=100) plt.xlabel("Abs(pt)")''' plt.figure(3) #plt.hist(del_R_mu_other_b,bins=50) lkn.hist_2D(minimal_dR,pt_jet,xbins=80,ybins=80,xrange=(0,3), yrange=(400,600)) plt.title(r"$Jet$ $vs$ $\Delta$$R$") plt.xlabel(r"$\Delta$R") plt.ylabel(r"$Jet$ $p$$_t$") plt.figure(4) #plt.hist(del_R_mu_b,bins=50) lkn.hist_2D(minimal_dR,pt_top,xbins=80,ybins=80,xrange=(0,3), yrange=(400,600)) plt.title(r"$t$ $\bar{t}$ $vs$ $\Delta$$R$") plt.xlabel(r"$\Delta$R") plt.ylabel(r"$t$ $\bar{t}$ $p$$_t$") plt.show() #python2.7-32 assigning_jets.py small_file.txt