Skip to content

Instantly share code, notes, and snippets.

@jmduarte
Created May 23, 2018 18:08
Show Gist options
  • Save jmduarte/54b130628cb1fb0e1295c478a87914ab to your computer and use it in GitHub Desktop.
Save jmduarte/54b130628cb1fb0e1295c478a87914ab to your computer and use it in GitHub Desktop.
checking double-b output
import ROOT as rt
from DataFormats.FWLite import Events,Handle
import rootpy #hands down, a better version of PyROOT
import rootpy.plotting as plt
events = Events("step2.root")
handle = Handle("vector<pat::Jet>")
#print [i for i in dir(jets[0]) if 'disc' in i.lower()]
#print [i.first for i in jets[0].getPairDiscri()]
histograms = {
'DoubleB' : plt.Hist(100,0,1),
'DeepDoubleB' : plt.Hist(100,0,1),
}
for evt in events:
evt.getByLabel("slimmedJetsAK8", handle)
jets = handle.product()
for j in jets:
for i in j.getPairDiscri():
#print [i.first for i in jets[0].getPairDiscri()]
if i.first=='pfBoostedDoubleSecondaryVertexAK8BJetTags':
print i.first, i.second
histograms['DoubleB'].fill((i.second+1)/2.)
elif i.first=='pfDeepDoubleBJetTags:probH':
print i.first, i.second
histograms['DeepDoubleB'].fill(i.second)
histograms['DoubleB'].linecolor = 'blue'
histograms['DeepDoubleB'].linecolor = 'red'
canvas = plt.Canvas(600, 600)
histograms['DoubleB'].Draw('hist')
histograms['DeepDoubleB'].Draw('histsame')
canvas.Draw() #You need this to display inline
canvas.Print('canvas.pdf')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment