In [1]:
import pandas as pd
import seaborn as sns

from nnfwtbn import Variable, Process, Cut, hist, HistogramFactory, McStack, DataStack
from nnfwtbn import toydata
In [2]:
df = toydata.get()
In [3]:
p_ztt = Process(r"$Z\rightarrow\tau\tau$", range=(0, 0))
p_sig = Process(r"Signal", range=(1, 1))
p_asimov = Process(r"Asimov", selection=lambda d: d.fpid >= 0)
In [4]:
s_bkg = McStack(p_ztt, p_sig)
s_data = DataStack(p_asimov)
In [5]:
c_vbf = Cut(lambda d: d.dijet_p4__M  > 400) & \
        Cut(lambda d: d.jet_1_p4__Pt >= 30) & \
        Cut(lambda d: d.is_dijet_centrality == 1) & \
        Cut(lambda d: d.jet_0_p4__Eta * df.jet_1_p4__Eta < 0) & \
        Cut(lambda d: (d.jet_0_p4__Eta - df.jet_1_p4__Eta).abs() > 3)
In [6]:
hist_factory = HistogramFactory(df, stacks=[s_bkg, s_data], bins=20, range=(0, 200), selection=None,
     weight="weight")
None
In [7]:
v_mmc = Variable(r"$m^H$", "higgs_m", "GeV")
hist_factory(v_mmc)
None
In [8]:
v_tau_pT = Variable(r"$p_\mathrm{T}{\tau}$", "tau_pt", "GeV")
hist_factory(v_tau_pT, bins=12, range=(0, 120))
None
In [9]:
v_lep_pT = Variable(r"$p_\mathrm{T}{\ell}$", "lep_pt", "GeV")
hist_factory(v_lep_pT, bins=12, range=(0, 120))
None
In [ ]: