{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"\n",
" ![](\"../../images/ATLASOD.gif\")
\n",
""
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" Notebook example: simple two-samples comparison
\n",
"\n",
"##### The following analysis is comparing the kinematics between events coming for the SM Higgs boson decaying to 2 W-bosons to those coming from the SM WW-diboson background production.\n",
"\n",
"SM Higgs to WW Feynman diagram:\n",
"![](\"../../images/fig_01a.png\")
\n",
"\n",
"SM WW-diboson Feynman diagram:\n",
"![](\"../../images/fig1b.png\")
"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Welcome to JupyROOT 6.18/04\n"
]
}
],
"source": [
"import ROOT"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"%jsroot on"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"3409043"
]
},
"execution_count": 3,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"## reading the input files via internet (URL to the file)\n",
"\n",
"## WW\n",
"bkg = ROOT.TFile.Open(\"https://atlas-opendata.web.cern.ch/atlas-opendata/samples/2020/2lep/MC/mc_363492.llvv.2lep.root\")\n",
"t_bkg = bkg.Get(\"mini\")\n",
"t_bkg.GetEntries()\n"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"628685"
]
},
"execution_count": 4,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"## SM H->WW\n",
"sig = ROOT.TFile.Open(\"https://atlas-opendata.web.cern.ch/atlas-opendata/samples/2020/2lep/MC/mc_345324.ggH125_WW2lep.2lep.root\")\n",
"t_sig = sig.Get(\"mini\")\n",
"t_sig.GetEntries()\n"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"c = ROOT.TCanvas(\"testCanvas\",\"a first way to plot a variable\",800,600)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
"h_bgs = ROOT.TH1F(\"h_bgs\",\"Example plot: Missing transverse energey\",20,0,200)\n",
"h2_bgs = ROOT.TH1F(\"h2_bgs\",\"Example plot: Number of Jets\",10,0,10)\n",
"\n",
"h_sig = ROOT.TH1F(\"h_sig\",\"Example plot: Missing transverse energey\",20,0,200)\n",
"h2_sig = ROOT.TH1F(\"h2_sig\",\"Example plot: Number of Jets\",10,0,10)"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {
"scrolled": true
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"10000\n",
"20000\n",
"30000\n",
"40000\n",
"50000\n",
"60000\n",
"70000\n",
"80000\n",
"90000\n",
"100000\n",
"110000\n",
"120000\n",
"130000\n",
"140000\n",
"150000\n",
"160000\n",
"170000\n",
"180000\n",
"190000\n",
"200000\n",
"210000\n",
"220000\n",
"230000\n",
"240000\n",
"250000\n",
"260000\n",
"270000\n",
"280000\n",
"290000\n",
"300000\n",
"310000\n",
"320000\n",
"330000\n",
"340000\n",
"350000\n",
"360000\n",
"370000\n",
"380000\n",
"390000\n",
"400000\n",
"410000\n",
"420000\n",
"430000\n",
"440000\n",
"450000\n",
"460000\n",
"470000\n",
"480000\n",
"490000\n",
"500000\n",
"510000\n",
"520000\n",
"530000\n",
"540000\n",
"550000\n",
"560000\n",
"570000\n",
"580000\n",
"590000\n",
"600000\n",
"610000\n",
"620000\n",
"630000\n",
"640000\n",
"650000\n",
"660000\n",
"670000\n",
"680000\n",
"690000\n",
"700000\n",
"710000\n",
"720000\n",
"730000\n",
"740000\n",
"750000\n",
"760000\n",
"770000\n",
"780000\n",
"790000\n",
"800000\n",
"810000\n",
"820000\n",
"830000\n",
"840000\n",
"850000\n",
"860000\n",
"870000\n",
"880000\n",
"890000\n",
"900000\n",
"910000\n",
"920000\n",
"930000\n",
"940000\n",
"950000\n",
"960000\n",
"970000\n",
"980000\n",
"990000\n",
"1000000\n",
"1010000\n",
"1020000\n",
"1030000\n",
"1040000\n",
"1050000\n",
"1060000\n",
"1070000\n",
"1080000\n",
"1090000\n",
"1100000\n",
"1110000\n",
"1120000\n",
"1130000\n",
"1140000\n",
"1150000\n",
"1160000\n",
"1170000\n",
"1180000\n",
"1190000\n",
"1200000\n",
"1210000\n",
"1220000\n",
"1230000\n",
"1240000\n",
"1250000\n",
"1260000\n",
"1270000\n",
"1280000\n",
"1290000\n",
"1300000\n",
"1310000\n",
"1320000\n",
"1330000\n",
"1340000\n",
"1350000\n",
"1360000\n",
"1370000\n",
"1380000\n",
"1390000\n",
"1400000\n",
"1410000\n",
"1420000\n",
"1430000\n",
"1440000\n",
"1450000\n",
"1460000\n",
"1470000\n",
"1480000\n",
"1490000\n",
"1500000\n",
"1510000\n",
"1520000\n",
"1530000\n",
"1540000\n",
"1550000\n",
"1560000\n",
"1570000\n",
"1580000\n",
"1590000\n",
"1600000\n",
"1610000\n",
"1620000\n",
"1630000\n",
"1640000\n",
"1650000\n",
"1660000\n",
"1670000\n",
"1680000\n",
"1690000\n",
"1700000\n",
"1710000\n",
"1720000\n",
"1730000\n",
"1740000\n",
"1750000\n",
"1760000\n",
"1770000\n",
"1780000\n",
"1790000\n",
"1800000\n",
"1810000\n",
"1820000\n",
"1830000\n",
"1840000\n",
"1850000\n",
"1860000\n",
"1870000\n",
"1880000\n",
"1890000\n",
"1900000\n",
"1910000\n",
"1920000\n",
"1930000\n",
"1940000\n",
"1950000\n",
"1960000\n",
"1970000\n",
"1980000\n",
"1990000\n",
"2000000\n",
"2010000\n",
"2020000\n",
"2030000\n",
"2040000\n",
"2050000\n",
"2060000\n",
"2070000\n",
"2080000\n",
"2090000\n",
"2100000\n",
"2110000\n",
"2120000\n",
"2130000\n",
"2140000\n",
"2150000\n",
"2160000\n",
"2170000\n",
"2180000\n",
"2190000\n",
"2200000\n",
"2210000\n",
"2220000\n",
"2230000\n",
"2240000\n",
"2250000\n",
"2260000\n",
"2270000\n",
"2280000\n",
"2290000\n",
"2300000\n",
"2310000\n",
"2320000\n",
"2330000\n",
"2340000\n",
"2350000\n",
"2360000\n",
"2370000\n",
"2380000\n",
"2390000\n",
"2400000\n",
"2410000\n",
"2420000\n",
"2430000\n",
"2440000\n",
"2450000\n",
"2460000\n",
"2470000\n",
"2480000\n",
"2490000\n",
"2500000\n",
"2510000\n",
"2520000\n",
"2530000\n",
"2540000\n",
"2550000\n",
"2560000\n",
"2570000\n",
"2580000\n",
"2590000\n",
"2600000\n",
"2610000\n",
"2620000\n",
"2630000\n",
"2640000\n",
"2650000\n",
"2660000\n",
"2670000\n",
"2680000\n",
"2690000\n",
"2700000\n",
"2710000\n",
"2720000\n",
"2730000\n",
"2740000\n",
"2750000\n",
"2760000\n",
"2770000\n",
"2780000\n",
"2790000\n",
"2800000\n",
"2810000\n",
"2820000\n",
"2830000\n",
"2840000\n",
"2850000\n",
"2860000\n",
"2870000\n",
"2880000\n",
"2890000\n",
"2900000\n",
"2910000\n",
"2920000\n",
"2930000\n",
"2940000\n",
"2950000\n",
"2960000\n",
"2970000\n",
"2980000\n",
"2990000\n",
"3000000\n",
"3010000\n",
"3020000\n",
"3030000\n",
"3040000\n",
"3050000\n",
"3060000\n",
"3070000\n",
"3080000\n",
"3090000\n",
"3100000\n",
"3110000\n",
"3120000\n",
"3130000\n",
"3140000\n",
"3150000\n",
"3160000\n",
"3170000\n",
"3180000\n",
"3190000\n",
"3200000\n",
"3210000\n",
"3220000\n",
"3230000\n",
"3240000\n",
"3250000\n",
"3260000\n",
"3270000\n",
"3280000\n",
"3290000\n",
"3300000\n",
"3310000\n",
"3320000\n",
"3330000\n",
"3340000\n",
"3350000\n",
"3360000\n",
"3370000\n",
"3380000\n",
"3390000\n",
"3400000\n",
"10000\n",
"20000\n",
"30000\n",
"40000\n",
"50000\n",
"60000\n",
"70000\n",
"80000\n",
"90000\n",
"100000\n",
"110000\n",
"120000\n",
"130000\n",
"140000\n",
"150000\n",
"160000\n",
"170000\n",
"180000\n",
"190000\n",
"200000\n",
"210000\n",
"220000\n",
"230000\n",
"240000\n",
"250000\n",
"260000\n",
"270000\n",
"280000\n",
"290000\n",
"300000\n",
"310000\n",
"320000\n",
"330000\n",
"340000\n",
"350000\n",
"360000\n",
"370000\n",
"380000\n",
"390000\n",
"400000\n",
"410000\n",
"420000\n",
"430000\n",
"440000\n",
"450000\n",
"460000\n",
"470000\n",
"480000\n",
"490000\n",
"500000\n",
"510000\n",
"520000\n",
"530000\n",
"540000\n",
"550000\n",
"560000\n",
"570000\n",
"580000\n",
"590000\n",
"600000\n",
"610000\n",
"620000\n",
"Done!\n"
]
}
],
"source": [
"n=0\n",
"for event in t_bkg:\n",
" n += 1\n",
" ## printing the evolution in number of events\n",
" if(n%10000==0):\n",
" print(n)\n",
" h_bgs.Fill((t_bkg.met_et)/1000.)\n",
" h2_bgs.Fill(t_bkg.jet_n)\n",
"\n",
"m=0 \n",
"for event in t_sig:\n",
" m += 1\n",
" ## printing the evolution in number of events\n",
" if(m%10000==0):\n",
" print(m)\n",
" h_sig.Fill((t_sig.met_et)/1000.)\n",
" h2_sig.Fill(t_sig.jet_n)\n",
" \n",
"print(\"Done!\")"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"scale_bgs = h_bgs.Integral()\n",
"h_bgs.Scale(1/scale_bgs)\n",
"\n",
"scale_sig = h_sig.Integral()\n",
"h_sig.Scale(1/scale_sig)\n",
"\n",
"\n",
"h_bgs.SetFillStyle(3001)\n",
"h_bgs.SetFillColor(4)\n",
"h_bgs.SetLineColor(4)\n",
"\n",
"h_sig.SetFillStyle(3003)\n",
"h_sig.SetFillColor(2)\n",
"h_sig.SetLineColor(2)\n",
"\n",
"legend=ROOT.TLegend(0.5,0.7,0.9,0.9)\n",
"legend.AddEntry(h_bgs,\"Background (WW) \",\"l\")\n",
"legend.AddEntry(h_sig,\"Signal (H #rightarrow WW)\",\"l\")\n",
"\n",
"h_sig.SetStats(0)\n",
"h_bgs.SetStats(0)\n",
"\n",
"h_sig.Draw(\"hist\")\n",
"h_bgs.Draw(\"histsame\")\n",
"legend.Draw()\n",
"c.Draw()\n"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"scale2_bgs = h2_bgs.Integral()\n",
"h2_bgs.Scale(1/scale2_bgs)\n",
"\n",
"scale2_sig = h2_sig.Integral()\n",
"h2_sig.Scale(1/scale2_sig)\n",
"\n",
"\n",
"\n",
"h2_bgs.SetFillStyle(3001)\n",
"h2_bgs.SetFillColor(4)\n",
"h2_bgs.SetLineColor(4)\n",
"\n",
"h2_sig.SetFillStyle(3003)\n",
"h2_sig.SetFillColor(2)\n",
"h2_sig.SetLineColor(2)\n",
"\n",
"legend=ROOT.TLegend(0.5,0.7,0.9,0.9)\n",
"legend.AddEntry(h2_bgs,\"Background (WW) \",\"l\")\n",
"legend.AddEntry(h2_sig,\"Signal (H #rightarrow WW)\",\"l\")\n",
"\n",
"\n",
"h2_sig.SetStats(0)\n",
"h2_bgs.SetStats(0)\n",
"h2_sig.Draw(\"hist\")\n",
"h2_bgs.Draw(\"histsame\")\n",
"legend.Draw()\n",
"c.Draw()\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Done**"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.6.9"
}
},
"nbformat": 4,
"nbformat_minor": 1
}