{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/home/hep/davide/miniconda3/envs/root_env/lib/ROOT.py:301: RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility\n",
" return _orig_ihook( name, *args, **kwds )\n"
]
}
],
"source": [
"import ROOT as r\n",
"import root_numpy as rn\n",
"import numpy as np\n",
"import root_numpy as rn\n",
"import matplotlib.pyplot as plt\n",
"import os\n",
"\n",
"from tools.data_processing import *\n",
"from tools.kinematics_fix import *\n",
"from tools.preselector import *\n"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"l_index = 0\n",
"mother_index = None\n",
"data_index = 0\n",
"n_cats=6\n",
"apply_phi_mass_cut = True\n",
"data_type = ['MC','data']\n",
"mother_ID=['Dplus','Ds']\n",
"l_flv = ['e','mu']\n",
"PATH='/disk/lhcb_data/davide/Rphipi_new/'\n",
"\n",
"save_path_presel_Dplus= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/presel/'+mother_ID[0]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_presel_Ds= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/presel/'+mother_ID[1]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"\n",
"save_path_truthed_Dplus= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/truthed/'+mother_ID[0]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_ghosts_Dplus= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/ghosts/'+mother_ID[0]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_truthed_Ds= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/truthed/'+mother_ID[1]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_ghosts_Ds= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/truthed/'+mother_ID[1]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"\n",
"save_path_trigged_Dplus = PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/trigged/'+mother_ID[0]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_trigged_Ds= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/trigged/'+mother_ID[1]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"\n",
"save_path_trigCats_Dplus= save_path_trigged_Dplus+'trigCats/'\n",
"save_path_trigCats_Ds= save_path_trigged_Ds+'trigCats/'\n",
"\n",
"save_path_BDT_for_Dplus= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/for_BDT_training/'+mother_ID[0]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'\n",
"save_path_BDT_for_Ds= PATH+data_type[data_index]+'/'+l_flv[l_index]+'_tuples/for_BDT_training/'+mother_ID[1]+'_phipi_'+l_flv[l_index]+l_flv[l_index]+'/'"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
"#retrieve files\n",
"\n",
"def find_file_path(l_index=l_index, mother_index=mother_index): \n",
" return \"/disk/lhcb_data/davide/Rphipi/MC/\"+mother_ID[mother_index]+\"_phipi_\"+l_flv[l_index]+l_flv[l_index]+\"/\"+mother_ID[mother_index]+\"_phipi_\"+l_flv[l_index]+l_flv[l_index]+\".root\"\n",
"\n",
"MC_Dplus = r.TFile(find_file_path(l_index=l_index, mother_index=0))\n",
"MC_Ds = r.TFile(find_file_path(l_index=l_index, mother_index=1))\n",
"\n",
"tree_name_Dplus = mother_ID[0]+'_OfflineTree/DecayTree'\n",
"tree_name_Ds = mother_ID[1]+'_OfflineTree/DecayTree'\n",
"\n",
"t_MC_Dplus = MC_Dplus.Get(tree_name_Dplus)\n",
"t_MC_Ds = MC_Ds.Get(tree_name_Ds)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Converting tuples to dictionary"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [],
"source": [
"#Switch on only the branches that you need\n",
"t_MC_Dplus.SetBranchStatus(\"*\",0)\n",
"\n",
"for branch in return_branches(data_index=data_index, mother_index=0, l_index=l_index):\n",
" if branch!='cos_thetal':\n",
" t_MC_Dplus.SetBranchStatus(branch, 1)\n",
" \n",
"MC_Dplus_tuple_dict_presel = {}\n",
"branches_needed=return_branches(data_index=data_index, mother_index=0, l_index=l_index)\n",
"\n",
"for branch in branches_needed:\n",
" if branch!='cos_thetal':\n",
" MC_Dplus_tuple_dict_presel[branch] = rn.root2array(\n",
" \n",
" filenames= find_file_path(l_index=l_index, mother_index=0),\n",
" treename = tree_name_Dplus,\n",
" branches = branch,\n",
" start=0,\n",
" stop=t_MC_Dplus.GetEntries(),\n",
" )\n",
" \n",
"t_MC_Ds.SetBranchStatus(\"*\",0)\n",
" \n",
"for branch in return_branches(data_index=data_index, mother_index=1, l_index=l_index):\n",
" if branch!='cos_thetal':\n",
" t_MC_Ds.SetBranchStatus(branch, 1)\n",
"\n",
"MC_Ds_tuple_dict_presel = {}\n",
"branches_needed=return_branches(data_index=data_index, mother_index=1, l_index=l_index)\n",
"for branch in branches_needed:\n",
" if branch!='cos_thetal':\n",
" MC_Ds_tuple_dict_presel[branch] = rn.root2array(\n",
" \n",
" filenames=find_file_path(l_index=l_index, mother_index=1),\n",
" treename = tree_name_Ds,\n",
" branches = branch,\n",
" start=0,\n",
" stop=t_MC_Ds.GetEntries(),\n",
" ) \n",
"\n",
"\n",
"MC_Dplus_tuple_dict_presel= fix_mass_vector(MC_Dplus_tuple_dict_presel, 'Dplus_ConsD_M')\n",
"MC_Ds_tuple_dict_presel = fix_mass_vector(MC_Ds_tuple_dict_presel, 'Ds_ConsD_M')\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Add Cos_thetal branch"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"\n",
"costheta_Dplus=get_costheta_list(MC_Dplus_tuple_dict_presel, l_index=l_index, mother_index=0)\n",
"costheta_Ds=get_costheta_list(MC_Ds_tuple_dict_presel, l_index=l_index, mother_index=1)\n",
"\n",
"MC_Dplus_tuple_dict_presel[\"cos_thetal\"]=costheta_Dplus\n",
"MC_Ds_tuple_dict_presel[\"cos_thetal\"]=costheta_Ds\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Save the tuples"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"presel\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/presel/Dplus_phipi_ee/Dplus_phipi_ee_presel.pickle\n",
"presel\n",
"Saving root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/presel/Ds_phipi_ee/Dplus_phipi_ee_presel.root\n",
"presel\n",
"Saving pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/presel/Dplus_phipi_ee/Ds_phipi_ee_presel.pickle\n",
"presel\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/presel/Ds_phipi_ee/Ds_phipi_ee_presel.root\n"
]
}
],
"source": [
"save_tuples(locals(), MC_Dplus_tuple_dict_presel, l_index, save_path_presel_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_presel, l_index, save_path_presel_Ds, file_format='root')\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_presel, l_index, save_path_presel_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_presel, l_index, save_path_presel_Ds, file_format='root')"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Truthmatching and saving the tuples"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"truthed\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Dplus_phipi_ee/Dplus_phipi_ee_truthed.pickle\n",
"truthed\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Dplus_phipi_ee/Dplus_phipi_ee_truthed.root\n",
"ghosts\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/ghosts/Dplus_phipi_ee/Dplus_phipi_ee_ghosts.pickle\n",
"ghosts\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/ghosts/Dplus_phipi_ee/Dplus_phipi_ee_ghosts.root\n",
"truthed\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Ds_phipi_ee/Ds_phipi_ee_truthed.pickle\n",
"truthed\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Ds_phipi_ee/Ds_phipi_ee_truthed.root\n",
"ghosts\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Ds_phipi_ee/Ds_phipi_ee_ghosts.pickle\n",
"ghosts\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/truthed/Ds_phipi_ee/Ds_phipi_ee_ghosts.root\n"
]
}
],
"source": [
"\n",
"MC_Dplus_tuple_dict_truthed = filter_tuple_cond(locals(), MC_Dplus_tuple_dict_presel, l_index, 'Dplus_BKGCAT', cmp_eq, 0)\n",
"MC_Dplus_tuple_dict_ghosts =filter_tuple_cond(locals(), MC_Dplus_tuple_dict_presel, l_index, 'Dplus_BKGCAT', cmp_eq, 60)\n",
"\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_truthed, l_index, save_path_truthed_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_truthed, l_index, save_path_truthed_Dplus, file_format='root')\n",
"\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_ghosts, l_index, save_path_ghosts_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_ghosts, l_index, save_path_ghosts_Dplus, file_format='root')\n",
"\n",
"\n",
"MC_Ds_tuple_dict_truthed = filter_tuple_cond(locals(), MC_Ds_tuple_dict_presel, l_index, 'Ds_BKGCAT', cmp_eq, 0)\n",
"MC_Ds_tuple_dict_ghosts = filter_tuple_cond(locals(), MC_Ds_tuple_dict_presel, l_index, 'Ds_BKGCAT', cmp_eq, 60)\n",
"\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_truthed, l_index, save_path_truthed_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_truthed, l_index, save_path_truthed_Ds, file_format='root')\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_ghosts, l_index, save_path_ghosts_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_ghosts, l_index, save_path_ghosts_Ds, file_format='root')\n",
" "
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Phi Mass Cut "
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {},
"outputs": [],
"source": [
"if apply_phi_mass_cut:\n",
" \n",
" if l_flv[l_index]=='mu':\n",
" lower_phi_mass = 990\n",
" upper_phi_mass = 1050\n",
" \n",
" if l_flv[l_index]=='e':\n",
" lower_phi_mass = 870\n",
" upper_phi_mass = 1110\n",
" \n",
" MC_Dplus_tuple_dict_truthed =filter_tuple_mass_cut(locals(), MC_Dplus_tuple_dict_truthed, l_index, 'phi_M', lower_bound=lower_phi_mass, upper_bound=upper_phi_mass, cut_kind='inside')\n",
" MC_Ds_tuple_dict_truthed =filter_tuple_mass_cut(locals(), MC_Ds_tuple_dict_truthed, l_index, 'phi_M', lower_bound=lower_phi_mass, upper_bound=upper_phi_mass, cut_kind='inside')\n",
" \n",
"#for i in range(20):\n",
"# \n",
"# upper_cut=np.arange(800+5*i,1100-5*i).max()\n",
"# lower_cut=np.arange(800+5*i,1100-5*i).min()\n",
"# plt.clf()\n",
"# MC_Dplus_tuple_dict_truthed_phi_cut=filter_tuple_mass_cut(globals(), MC_Dplus_tuple_dict_truthed, l_index, 'phi_M', lower_bound=lower_cut, upper_bound=upper_cut)\n",
"# plt.hist(MC_Dplus_tuple_dict_truthed_phi_cut['Dplus_ConsD_M'],bins=40);\n",
"# plt.savefig('plot_{0}.png'.format(i))"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Get Hlt cuts "
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Hlt1\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_Hlt1.pickle\n",
"Hlt1\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_Hlt1.root\n",
"Hlt1\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_Hlt1.pickle\n",
"Hlt1\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_Hlt1.root\n",
"Hlt2\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_Hlt2.pickle\n",
"Hlt2\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_Hlt2.root\n",
"Hlt2\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_Hlt2.pickle\n",
"Hlt2\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_Hlt2.root\n",
"FullHlt\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_FullHlt.pickle\n",
"FullHlt\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/Dplus_phipi_ee_FullHlt.root\n",
"FullHlt\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_FullHlt.pickle\n",
"FullHlt\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/Ds_phipi_ee_FullHlt.root\n"
]
}
],
"source": [
"#Hlt1\n",
"MC_Dplus_tuple_dict_Hlt1 = filter_tuple_cond(locals(), MC_Dplus_tuple_dict_truthed, l_index, 'Dplus_Hlt1TrackMVADecision_TOS', cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_Hlt1, l_index, save_path_trigged_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_Hlt1, l_index, save_path_trigged_Dplus, file_format='root')\n",
"\n",
"MC_Ds_tuple_dict_Hlt1 = filter_tuple_cond(locals(), MC_Ds_tuple_dict_truthed, l_index, 'Ds_Hlt1TrackMVADecision_TOS', cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_Hlt1, l_index, save_path_trigged_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_Hlt1, l_index, save_path_trigged_Ds, file_format='root')\n",
"\n",
"#Hlt2\n",
"MC_Dplus_tuple_dict_Hlt2 = filter_tuple_cond(locals(), MC_Dplus_tuple_dict_truthed, l_index, \"Dplus_Hlt2RareCharmD2Pi\"+l_flv[l_index].capitalize()+l_flv[l_index].capitalize()+\"OSDecision_TOS\",cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_Hlt2, l_index, save_path_trigged_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_Hlt2, l_index, save_path_trigged_Dplus, file_format='root')\n",
"\n",
"MC_Ds_tuple_dict_Hlt2 = filter_tuple_cond(locals(), MC_Ds_tuple_dict_truthed, l_index, \"Ds_Hlt2RareCharmD2Pi\"+l_flv[l_index].capitalize()+l_flv[l_index].capitalize()+\"OSDecision_TOS\",cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_Hlt2, l_index, save_path_trigged_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_Hlt2, l_index, save_path_trigged_Ds, file_format='root')\n",
"\n",
"#FullHlt\n",
"MC_Dplus_tuple_dict_FullHlt = filter_tuple_cond(locals(), MC_Dplus_tuple_dict_Hlt1, l_index, \"Dplus_Hlt2RareCharmD2Pi\"+l_flv[l_index].capitalize()+l_flv[l_index].capitalize()+\"OSDecision_TOS\",cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_FullHlt, l_index, save_path_trigged_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_FullHlt, l_index, save_path_trigged_Dplus, file_format='root')\n",
"\n",
"MC_Ds_tuple_dict_FullHlt = filter_tuple_cond(locals(), MC_Ds_tuple_dict_Hlt1, l_index, \"Ds_Hlt2RareCharmD2Pi\"+l_flv[l_index].capitalize()+l_flv[l_index].capitalize()+\"OSDecision_TOS\",cmp_eq, True)\n",
"\n",
"save_tuples(locals(), MC_Ds_tuple_dict_FullHlt, l_index, save_path_trigged_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_FullHlt, l_index, save_path_trigged_Ds, file_format='root')\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Split in trigger categories\n",
"\n",
" if trig_cat == 0:\n",
" #Exclusive L0_Lepton_TOS \n",
" \n",
" if trig_cat == 1:\n",
" #Exclusive L0_Global_TIS\n",
" \n",
" if trig_cat == 2:\n",
" #L0_Lepton_TOS and L0_Global_TIS\n",
" \n",
" if trig_cat == 3:\n",
" #Either L0_Lepton_TOS or L0_Global_TIS\n",
" \n",
" if trig_cat == 4:\n",
" #TOS and TOSandTIS\n",
"\n",
" if trig_cat == 5:\n",
" #TIS and TOSandTIS\n",
"\n",
" \n",
" "
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"The counts come back\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/0/Dplus_phipi_ee_trigCat0.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/1/Dplus_phipi_ee_trigCat1.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/2/Dplus_phipi_ee_trigCat2.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/3/Dplus_phipi_ee_trigCat3.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/4/Dplus_phipi_ee_trigCat4.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/5/Dplus_phipi_ee_trigCat5.pickle\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/0/Dplus_phipi_ee_trigCat0.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/1/Dplus_phipi_ee_trigCat1.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/2/Dplus_phipi_ee_trigCat2.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/3/Dplus_phipi_ee_trigCat3.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/4/Dplus_phipi_ee_trigCat4.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Dplus_phipi_ee/trigCats/5/Dplus_phipi_ee_trigCat5.root\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/0/Ds_phipi_ee_trigCat0.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/1/Ds_phipi_ee_trigCat1.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/2/Ds_phipi_ee_trigCat2.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/3/Ds_phipi_ee_trigCat3.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/4/Ds_phipi_ee_trigCat4.pickle\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/5/Ds_phipi_ee_trigCat5.pickle\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/0/Ds_phipi_ee_trigCat0.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/1/Ds_phipi_ee_trigCat1.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/2/Ds_phipi_ee_trigCat2.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/3/Ds_phipi_ee_trigCat3.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/4/Ds_phipi_ee_trigCat4.root\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/trigged/Ds_phipi_ee/trigCats/5/Ds_phipi_ee_trigCat5.root\n"
]
}
],
"source": [
"MC_Dplus_tuple_dict_trigCats = {}\n",
"MC_Ds_tuple_dict_trigCats = {}\n",
"\n",
"for i in range(n_cats):\n",
" \n",
" MC_Dplus_tuple_dict_trigCats[i] = filter_tuple_L0TrigCat(locals(), MC_Dplus_tuple_dict_FullHlt, l_index, i)\n",
" MC_Ds_tuple_dict_trigCats[i] = filter_tuple_L0TrigCat(locals(), MC_Ds_tuple_dict_FullHlt, l_index, i)\n",
" \n",
"n_0=MC_Dplus_tuple_dict_trigCats[0]['Dplus_ConsD_M'].shape[0]\n",
"n_1=MC_Dplus_tuple_dict_trigCats[1]['Dplus_ConsD_M'].shape[0]\n",
"n_2=MC_Dplus_tuple_dict_trigCats[2]['Dplus_ConsD_M'].shape[0]\n",
"n_3=MC_Dplus_tuple_dict_trigCats[3]['Dplus_ConsD_M'].shape[0]\n",
"n_4=MC_Dplus_tuple_dict_trigCats[4]['Dplus_ConsD_M'].shape[0]\n",
"n_5=MC_Dplus_tuple_dict_trigCats[5]['Dplus_ConsD_M'].shape[0]\n",
"\n",
"sanity_1=np.array_equal([n_0+n_2+n_1==n_3,n_0+n_5==n_3,n_4+n_1==n_3],[True,True,True])\n",
"\n",
"n_0=MC_Ds_tuple_dict_trigCats[0]['Ds_ConsD_M'].shape[0]\n",
"n_1=MC_Ds_tuple_dict_trigCats[1]['Ds_ConsD_M'].shape[0]\n",
"n_2=MC_Ds_tuple_dict_trigCats[2]['Ds_ConsD_M'].shape[0]\n",
"n_3=MC_Ds_tuple_dict_trigCats[3]['Ds_ConsD_M'].shape[0]\n",
"n_4=MC_Ds_tuple_dict_trigCats[4]['Ds_ConsD_M'].shape[0]\n",
"n_5=MC_Ds_tuple_dict_trigCats[5]['Ds_ConsD_M'].shape[0]\n",
"\n",
"sanity_2=np.array_equal([n_0+n_2+n_1==n_3,n_0+n_5==n_3,n_4+n_1==n_3],[True,True,True])\n",
"if sanity_1 and sanity_2:\n",
" print('The counts come back')\n",
" \n",
" save_tuples(locals(), MC_Dplus_tuple_dict_trigCats, l_index, save_path_trigCats_Dplus, file_format='pickle')\n",
" save_tuples(locals(), MC_Dplus_tuple_dict_trigCats, l_index, save_path_trigCats_Dplus, file_format='root')\n",
" \n",
" save_tuples(locals(), MC_Ds_tuple_dict_trigCats, l_index, save_path_trigCats_Ds, file_format='pickle')\n",
" save_tuples(locals(), MC_Ds_tuple_dict_trigCats, l_index, save_path_trigCats_Ds, file_format='root')\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Prepare Sig Sets for BDT"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"for_BDT\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/for_BDT/Dplus_phipi_ee/Dplus_phipi_ee_for_BDT.pickle\n",
"for_BDT\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/for_BDT/Dplus_phipi_ee/Dplus_phipi_ee_for_BDT.root\n",
"for_BDT\n",
"Overwriting pickle file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/for_BDT/Ds_phipi_ee/Ds_phipi_ee_for_BDT.pickle\n",
"for_BDT\n",
"Overwriting root file at/disk/lhcb_data/davide/Rphipi_new/MC/e_tuples/for_BDT/Ds_phipi_ee/Ds_phipi_ee_for_BDT.root\n"
]
}
],
"source": [
"if l_flv[l_index]=='mu':\n",
" \n",
" lower_Dplus_mass_mc=1830\n",
" upper_Dplus_mass_mc=1910\n",
" \n",
" lower_Ds_mass_mc=1930\n",
" upper_Ds_mass_mc=2010\n",
" \n",
" \n",
"if l_flv[l_index]=='e':\n",
" \n",
" lower_Dplus_mass_mc=1790\n",
" upper_Dplus_mass_mc=1950\n",
"\n",
" lower_Ds_mass_mc=1890\n",
" upper_Ds_mass_mc=2050\n",
" \n",
"\n",
"MC_Dplus_tuple_dict_sig_for_BDT = filter_tuple_mass_cut(locals(), MC_Dplus_tuple_dict_FullHlt, l_index, 'Dplus_ConsD_M', lower_bound=lower_Dplus_mass_mc, upper_bound=upper_Dplus_mass_mc, cut_kind='inside') \n",
"save_tuples(locals(), MC_Dplus_tuple_dict_sig_for_BDT, l_index, save_path_BDT_for_Dplus, file_format='pickle')\n",
"save_tuples(locals(), MC_Dplus_tuple_dict_sig_for_BDT, l_index, save_path_BDT_for_Dplus, file_format='root')\n",
"\n",
"\n",
"MC_Ds_tuple_dict_sig_for_BDT = filter_tuple_mass_cut(locals(), MC_Ds_tuple_dict_FullHlt, l_index, 'Ds_ConsD_M', lower_bound=lower_Ds_mass_mc, upper_bound=upper_Ds_mass_mc, cut_kind='inside') \n",
"save_tuples(locals(), MC_Ds_tuple_dict_sig_for_BDT, l_index, save_path_BDT_for_Ds, file_format='pickle')\n",
"save_tuples(locals(), MC_Ds_tuple_dict_sig_for_BDT, l_index, save_path_BDT_for_Ds, file_format='root')\n"
]
},
{
"cell_type": "code",
"execution_count": 12,
"metadata": {},
"outputs": [
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAYEAAAD8CAYAAACRkhiPAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADl0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uIDIuMi4yLCBodHRwOi8vbWF0cGxvdGxpYi5vcmcvhp/UCwAADkxJREFUeJzt3V+s5OVdx/H3p+1KUqsG2EOb9Ox6Lpo0Goqke0AbtKWKAdlEE7EQdPWCmhVT/1BvKPUCuSDhQjTBmtBNwCoYspGWNqa7Cr1CkibkgHT/sFj/hMIxhT0V03ChBMvXi/Ngx+Uczp+ZOWdmnvcrmcxvnvnNzPPdJzufeX7/TqoKSVKf3rHbHZAk7R5DQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktSxd+12BwD27t1bCwsLu90NSZoqTz311Heqam6Y95iIEFhYWGBpaWm3uyFJUyXJt4Z9DzcHSVLHDAFJ6pghIEkdMwQkqWOGgCR1zBCQpI4ZApLUMUNAkjpmCEhSxybijGFNroXPfPX/PX7+roO71BNJ4+BMQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktQxQ0CSOmYISFLHDAFJ6pghIEkdMwQkqWNeRVRb4lVFpdniTECSOmYISFLHNgyBJPuSPJ7kVJJvJrm1tV+Q5LEkJ5M8muT8gdfcluRMe83V4yxAkrR9m5kJvA78TlVdDBwAfjPJpcAdwPGq+hBwvD0myQHgOuAS4Brg80nOG0fnJUnD2TAEquqlqjrRll8FTgDvBw4CD7TVHmyPafdHq+r1qloGTgOXj7rjkqThbWmfQJIF4DLgCWCuqlYA2v1FbbV54MWBly23tnPf63CSpSRLKysrW++5JGlomz5ENMl7gIeBW6rqu0mG+uCqOgIcAVhcXKyh3kwjce7hn5Jm36ZmAkn2AF8EHqqqL7XmlSRz7fk54GxrXwb2Dbx8vrVJkibMZo4OCnAfcKaq7h546hhwqC0fYnXn8JvtNyTZk2QeuBh4cnRdliSNymY2B10B/DpwMskzre2zwO3A0SQ3AS8D1wNU1VKSR1jdgfwGcHNVvTbynkuShrZhCFTVE8B6OwCuWuc1dwJ3DtEvSdIO8IxhSeqYISBJHTMEJKljhoAkdcwQkKSOGQKS1DFDQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktQxQ0CSOmYISFLHDAFJ6pghIEkdMwQkqWOGgCR1zBCQpI4ZApLUMUNAkjpmCEhSxwwBSeqYISBJHTMEJKljhoAkdcwQkKSOGQKS1DFDQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjm0YAknuT3I2yamBtj9K8u9Jnmm3aweeuy3JmSSnklw9ro5Lkoa3mZnAF4Br1mj/06q6tN2OASQ5AFwHXNJe8/kk542qs5Kk0dowBKrqceCVTb7fQeBoVb1eVcvAaeDyIfonSRqjYfYJfCrJc0n+OsmFrW0eeHFgneXWJkmaQNsNgT8HPgD8OPCvwD1bfYMkh5MsJVlaWVnZZjckScPYVghU1UpVfa+q3gDuBS5rTy0D+wZWnW9ta73HkaparKrFubm57XRDkjSkbYVAkosGHl4HPNuWjwE3JNmTZB64GHhyuC5KksblXRutkOQh4Epgb5Jl4Hbg40kuAX4AeAH4JEBVLSV5BDgBvAHcXFWvjanvkqQhbRgCVXXjGs33vc36dwJ3DtMpSdLO2DAEpLez8JmvvqXt+bsO7kJPJG2Hl42QpI45E+jYWr/iJfXFmYAkdcwQkKSOGQKS1DFDQJI6ZghIUscMAUnqmCEgSR3zPAFJb3HuOSSeBT67nAlIUscMAUnqmJuDpM55+ZC+OROQpI4ZApLUMUNAkjpmCEhSxwwBSeqYISBJHfMQ0U54GKCktTgTkKSOGQKS1DFDQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktQxQ0CSOmYISFLHDAFJ6pghIEkdMwQkqWOGgCR1zBCQpI5tGAJJ7k9yNsmpgbYLkjyW5GSSR5OcP/DcbUnOJDmV5OpxdVySNLzN/HnJLwCfA/5qoO0O4HhV/UmST7fHv5fkAHAdcAnwXuCJJB+sqtdG221NsnP/lOXzdx3cpZ5I2siGM4Gqehx45Zzmg8ADbfnB9vjN9qNV9XpVLQOngctH1FdJ0ohtd5/AXFWtALT7i1r7PPDiwHrLrU2SNIE2szloLJIcBg4D7N+/f7e6IXXn3M116tt2ZwIrSeYA2v3Z1r4M7BtYb761vUVVHamqxapanJub22Y3JEnD2G4IHAMOteVDwPGB9huS7EkyD1wMPDlcFyVJ47Lh5qAkDwFXAnuTLAO3t9vRJDcBLwPXA1TVUpJHgBPAG8DNHhkkSZNrwxCoqhvXeeqqdda/E7hzmE5Jmiwe9ju7PGNYkjpmCEhSxwwBSerYrp0nIGl6rXWugfsJppMzAUnqmCEgSR0zBCSpY4aAJHXMEJCkjnl0kDTjvGqo3o4zAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktQxQ0CSOmYISFLHPFlsRk3SCUL+aUJpcjkTkKSOGQKS1DFDQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXMEJCkjhkCktQxQ0CSOmYISFLHDAFJ6pghIEkd81LSU2ity0R7eWZJ2+FMQJI6ZghIUsfcHCTNkEn6i3KaDkOFQJLngVeB7wH/U1WLSS4AjgLvA74N3FBV/zlsRyVJozeKzUEfr6pLq2qxPb4DOF5VHwKOt8eSpAk0jn0CB4EH2vKD7bEkaQINGwIFPJbkZJLfbW1zVbUC0O4vGvIzJEljMuyO4Y9U1UtJLgL+Lslzm31hksPAYYD9+/cP2Q1Ju+3cndKeuzIdhpoJVNVL7f4s8DBwGbCSZA6g3Z9d57VHqmqxqhbn5uaG6YYkaZu2HQJJfjDJu99cBq4BngWOAYfaaodY3TksSZpAw2wOei/w5SQFvJvVw0K/AvwDcDTJTcDLwPVD91Ib8vhwSdux7RCoqn8DLlnjqf8Artp2jyRJO8YzhqUp5gxQw/LaQZLUMUNAkjpmCEhSxwwBSeqYISBJHTMEJKljhoAkdczzBKaAx4JLGhdnApLUMWcCmghehljaHc4EJKljzgS049zHIU0OQ0DSWLiJbzoYApoKa80e/FKRhuc+AUnqmDOBCeQ2c/8NpJ3iTECSOuZMQNoBm9lJ6o5U7QZnApLUMUNAkjrm5iBNrR43n7jDXKNmCOwy/1NL2k1uDpKkjjkTkHaBM0BNCmcCktQxZwI7zF+A49PjjmJpWIaAtEWbCfJRBNCs/WDwIoCTyRCQBvhFpd64T0CSOuZMQDPLX/XSxgwBSRPDnfs7zxBQ18a183XWduruFmdz42cIqCvb+XL2C32yOFsYLUNghPzVImnaGAJj5q9IabwmaWYwSX3ZLENA0q4Zx4+knTqZb1YYAuvYzKYdf+VLmnZjC4Ek1wB/DLwT+MuqumtcnyVJWzGNm23GZSwhkOQ84F7gZ4CXgK8nebSqnh7H543CZn7V+8tf6teoDvyYtAAa10zgJ4HTVfUiQJKjwEFgV0LAo3YkjcMs/DAcVwjMAy8OPF4GrhzTZ01cskqaLj3/UExVjf5Nk18FPlpVN7fHNwJXVtVvDaxzGDjcHn4Q+KcRd2Mv8J0Rv+c06LVu6Ld26+7LYN0/WlVzw7zZuGYCy8C+gcfzre3/VNUR4MiYPp8kS1W1OK73n1S91g391m7dfRl13eO6lPSTwMVJ5pPsAW4Ajo/psyRJ2zSWmUBV/XeS3wb+ntWgebCqlsbxWZKk7RvbeQJVdQw4Nq7334SxbWqacL3WDf3Wbt19GWndY9kxLEmaDv55SUnq2NSEQJL7k5xNcmqg7Yok30hyOsmJJFcMPHdbkjNJTiW5eqD9QJJ/TPJsknuSZKdr2aqt1J5kIcl/JXmm3e4deM1U1b5O3YtJnm51/22SHx54bibGfCt1z9h470vyeBu/bya5tbVfkOSxJCeTPJrk/IHXTP2Yb7XukY95VU3FDfgo8GHg1EDbE8AvtOVrgSfa8gFgCdjD6uGpzwPntedOAAfa8leAX97t2kZc+8Lgeue8z1TVvk7dJ4GPteWbgLtnbcy3WPcsjff7gEva8g8B/wxcCvwZ8Aet/dPAPbM05tuoe6RjPjUzgap6HHjlnOZl4M1fgj8CvNCWDwJHq+r1qloGTgOXJ9kPvLOqnmrrPdjWnWhbrH1N01j7OnV/AHi8LT8G/GJbnpkx32Lda5rSul+qqhNt+VVWv9Dez2q/H2irDdYxE2O+jbrXtN26pyYE1nErcHeSF1m9YultrX2ty1bMv037NFqvdoCFtqno60l+rrXNSu1ngF9qy58A9rflWR/z9eqGGRzvJAvAZazOeOeqagWg3V/UVpu5Md9k3TDCMZ/2ELgP+P2q2sfqdOm+Xe7PTlqv9m8D81X1E8CngAcGt6HOgN8Abmnby/cCr+1yf3bKenXP3HgneQ/wMHBLVX13t/uzU7ZQ90jHfNr/qMxHgJ9vy38D/EVbXu+yFRtezmKKrFl7Vb1G+4Koqqfbl8aPMSO1V9Up2sUI26+ma9tTMz3m69U9a+Od1SsMfBF4qKq+1JpXksxV1UqSOeBsa5+ZMd9K3aMe82mfCXwL+Fhb/llWdwzB6klqNyTZk2QeuBh4sqpeAN5I8uG23q8xvZezWLP2JBcmeUdbXmC19n+ZldqT7G33AT7L92dAMz3m69U9S+PdarsPOFNVdw88dQw41JYP8f06ZmLMt1r3yMd8t/eMb2EP+kOsToNeZzXdPglcAXwDeBZ4BvipgfX/kNXtqKdpR9G09sW27rPA52gnzE3ybSu1A7/Saj4JnAI+Ma21r1P3LcBzrba7BmuYlTHfSt0zNt4/DRSrO0afabdrgQuBr7UavwZcMEtjvtW6Rz3mnjEsSR2b9s1BkqQhGAKS1DFDQJI6ZghIUscMAUnqmCEgSR0zBCSpY4aAJHXsfwE7/xiPvXWzDQAAAABJRU5ErkJggg==\n",
"text/plain": [
"<Figure size 432x288 with 1 Axes>"
]
},
"metadata": {
"needs_background": "light"
},
"output_type": "display_data"
}
],
"source": [
"plt.hist(np.concatenate([MC_Dplus_tuple_dict_sig_for_BDT['Dplus_ConsD_M'],MC_Ds_tuple_dict_sig_for_BDT['Ds_ConsD_M']]), bins=70);\n",
" \n",
" "
]
},
{
"cell_type": "code",
"execution_count": 13,
"metadata": {},
"outputs": [
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAXQAAAD8CAYAAABn919SAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADl0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uIDIuMi4yLCBodHRwOi8vbWF0cGxvdGxpYi5vcmcvhp/UCwAAESNJREFUeJzt3X2MZXddx/H3h1LAloLd9q4NtMuGxDTGtllhGiSolFBi2wUSgrYQlsQsZsEotTUoVKOl0T8wkYegJmR5iHGX1I0FQWEHuoQ0tRox27J0tw+UYIqFtLvTUgtCRB6+/jFnZdjO7D33zr13Zn77fiWT83DPOfc7Z+5+9nd/95zfTVUhSdr4nrLWBUiSJsNAl6RGGOiS1AgDXZIaYaBLUiMMdElqhIEuSY0w0CWpEQa6JDXiqbN8snPPPbe2bt06y6eUpA3vzjvvfLSqBsO2m2mgb926lYMHD87yKSVpw0vytT7b2eUiSY0w0CWpEQa6JDXCQJekRhjoktSIoYGe5MIkh5b8fCvJdUk2JTmQ5HCSW5OcPYuCJUnLGxroVfXlqtpWVduAFwLfBf4BuAmYr6qLgfluWZK0Rkbtcnk58NWq+hqwHdjTrd/bLUuS1siogf464OZuflBVCwDddPMkC5Mkjab3naJJnga8GrhhlCdIsgvYBbBly5aRipPWs63v+PST1j34Lt+oau2M0kK/Erirqo52ywtJBgDd9NhyO1XV7qqaq6q5wWDoUASSpDGNEuiv58fdLQD7gR3d/A4WPxiVJK2RXl0uSc4EXgG8ecnqG4F9SXYCR4GrJ1+eJKmvXoFeVd8Bzjlh3WPA5dMoSpI0Ou8UlaRGGOiS1AgDXZIaYaBLUiMMdElqhIEuSY0w0CWpEQa6JDXCQJekRhjoktQIA12SGmGgS1Ijen/BhaSN7cQv5PDLONpjC12SGmGgS1Ij7HKR1phdIZoUW+iS1AgDXZIaYZeLJubEroMTnYpdCXanaJZsoUtSIwx0SWpEry6XJD8NfBC4EHgasBO4H9gHnAc8DFxTVY9PqU5JE7ZcF5ldQhtb3xb6B4FPVNUlwEXAPcBNwHxVXQzMd8uSpDUyNNCTnAP8QlV9FKCqflBVTwDbgT3dZnu7ZUnSGunTQv9ZYCHJ3ye5J8meJGcBg6paAOimm5fbOcmuJAeTHFxYWJhc5ZKkn9An0J8CXAr8RVX9PPBN4I/7PkFV7a6quaqaGwwGY5YpSRqmT6A/BHyjqr7QLd8CbGOx1T4A6KbHplOiJKmPoVe5VNVDSR5NcmFVfRl4OYtXuDwA7ADe203np1qp1PFmHc+Bltf3TtE3AR9Ncgbwn8AbuvX7kuwEjgJXT6E+SVJPvQK9qg4Bc8s8dPlky5EkjcuxXLSurKeuhPVUi9SHt/5LUiMMdElqhF0u6mXY0LgbXeu/n04NttAlqREGuiQ1wi4XqSe7ZbTe2UKXpEYY6JLUCANdkhphoEtSIwx0SWqEV7low2t9zBWvrlFfttAlqREGuiQ1wkCXpEYY6JLUCANdkhrhVS5aU17BMT2e21OPLXRJakSvFnqSB4FvAz8EflBVc0k2AfuA84CHgWuq6vFpFSpJOrlRulxeVlWPLlm+CZivqvckub5bvnai1WnVlnvbvZFuvBmn26DP72x3hFq0mi6X7cCebn5vtyxJWiN9A72AA0kOJ3lrt25QVQsA3XTzNAqUJPXTt8vlxVX1SJLNwGeS3N/3CZLsAnYBbNmyZYwSTx2tj0myluxi0amgVwu9qh7ppseAW4BLgYUkA4BuemyFfXdX1VxVzQ0Gg8lULUl6kqGBnuTMJGccnweuAO4F9gM7us12APPTKlKSNFyfLpefAT6RpIAzWLxU8ZPAPwP7kuwEjgJXT63KU5RdMKcmu4c0rqGBXlX/AVyyzEOPAZdPvCJJ0li8U1SSGuFYLnoS3/JLG5MtdElqhIEuSY2wy0WaIbuzNE220CWpEQa6JDXCLpcNZKMPhXsqsEtFa8kWuiQ1wkCXpEbY5SK7CaRG2EKXpEYY6JLUCLtcpAZMqtvMIZs3NlvoktQIA12SGmGgS1IjDHRJaoSBLkmNMNAlqREGuiQ1onegJzktyReTfKpb3pTkQJLDSW5Ncvb0ypQkDTPKjUW/C9wHPKtbvgmYr6r3JLm+W752wvWtC95sIS1yCOf1rVcLPcn5wHbgQ0tWbwf2dPN7u2VJ0hrp20J/H/AHwFlL1g2qagGgqhaSbF5uxyS7gF0AW7ZsWUWpkmbNkTg3lqEt9CSvBI5V1Z3jPEFV7a6quaqaGwwG4xxCktRDnxb6S4BXJ7kKeAbwrCR7gYUkg651PgCOTbNQSdLJDQ30qroBuAEgyWXA26pqR5K/AnYA7+2m81Osc13Z6B+S+jZakzTOv4eN/m9ovVrN8Lk3AvuS7ASOAldPpiRJ0jhGCvSqug24rZt/DLh88iVJksbhF1w0xu4UbUR2wUyGt/5LUiMMdElqhF0uG9xG6mLZSLVqcvy7z44tdElqhIEuSY2wy0XSRNnFsnZsoUtSIwx0SWqEgS5JjTDQJakRBrokNcKrXNaIVwJIK/O7S8djC12SGmGgS1IjDHRJaoSBLkmNMNAlqRFe5TIjXtUiTZbfcvRkttAlqRFDAz3JM5IcTHIoyVeSvC+LNiU5kORwkluTnD2LgiVJy+vT5fI94KVV9Z0kpwN3AC8DXgPMV9V7klwP3ARcO71S1y9vgpC0Hgxtodei73SLpwOnAceA7cCebv3eblmStEZ69aEnOS3JIRaD/LaqOgIMqmoBoJtunl6ZkqRhegV6Vf2wqrYB5wO/nORlfZ8gya6uD/7gwsLCuHVKkoYY6SqXqvov4NPALwILSQYA3fTYCvvsrqq5qpobDAarrVeStII+V7mcm+Ssbv6ngFcAR4D9wI5usx3A/LSKlCQN1+cql+cAf5skwDOAm6vqn5L8K7AvyU7gKHD1FOucGm9OkNrg1WY9Ar2q7ga2LbP+MeDyaRQlSRqdd4pKUiMcy0XShuB4SMPZQpekRhjoktQIu1ymxLeHkmbNFrokNcJAl6RGGOiS1AgDXZIaYaBLUiNOuatcvPpE0nGtjf9iC12SGmGgS1IjDHRJaoSBLkmNMNAlqREGuiQ1wkCXpEYY6JLUCANdkhphoEtSI4YGepILktye5EiSB5K8vVu/KcmBJIeT3Jrk7OmXK0laSZ8W+veB36mqi4AXAr+ZZBtwEzBfVRcD892yJGmNDA30qnqkqu7u5r8N3A08F9gO7Ok229stS5LWyEh96Em2ApcCdwCDqloA6KabJ12cJKm/3sPnJnkmcAtwXVU9kaTvfruAXQBbtmwZp0ZJGsuJw+Nu5KFx++jVQk9yOvAx4Oaq+ni3eiHJoHt8ABxbbt+q2l1Vc1U1NxgMJlGzJGkZfa5yCfBh4L6qeveSh/YDO7r5HSx+MCpJWiN9ulxeArwROJzkULfuD4EbgX1JdgJHgaunU+Js+Y1GkjaqoYFeVXcAK3WYXz7ZciRJ4/JOUUlqRNNfEm33iaTV2khXythCl6RGGOiS1AgDXZIaYaBLUiMMdElqhIEuSY0w0CWpEU1fhy5JS7V+b4otdElqhIEuSY0w0CWpEQa6JDXCQJekRhjoktQIA12SGmGgS1IjDHRJaoSBLkmNMNAlqRFDx3JJ8hHglcCxqrqoW7cJ2AecBzwMXFNVj0+zUEmahY083kufFvrfAFecsO4mYL6qLgbmu2VJ0hoaGuhVdTvwzRNWbwf2dPN7u2VJ0hoatw99UFULAN1080obJtmV5GCSgwsLC2M+nSRpmKl/KFpVu6tqrqrmBoPBtJ9Okk5Z4wb6QpIBQDc9NrmSJEnjGPcbi/YDO4D3dtP5iVU0ghM/jX7wXXblS5qu5a6CWS/Z0+eyxZuBy4Bzk3wduLH72ZdkJ3AUuHqaRUqShhsa6FX1+hUeunzCtUiSVmHDfEl0n4v9N/INAZK0Wt76L0mNMNAlqREbpstFktar9XLFnS10SWqEgS5JjTDQJakRBrokNcJAl6RGGOiS1AgDXZIaYaBLUiMMdElqhIEuSY0w0CWpEY7lIkkTtlbfamQLXZIaYaBLUiMMdElqhIEuSY1YVaAnuSLJkST3JXnHpIqSJI1u7EBP8nTgA8CVwCXAryV5waQKkySNZjUt9BcB91TVQ1X1fWAfsDbfuyRJWlWgnw88tGT56906SdIamPqNRUl2Abu6xf9O8uURdj8XeHTyVU2EtY1uvdYF1jaO9VoXrMPa8uf/PztObc/rs9FqAv3rwAVLls/v1v2EqtoN7B7nCZIcrKq58cqbLmsb3XqtC6xtHOu1Ljh1a1tNl8u/AxclOT/J6cA1wPxkypIkjWrsFnpV/U+S3wI+y+J/DHur6uDEKpMkjWRVfehVtR/YP6FaljNWV82MWNvo1mtdYG3jWK91wSlaW6pqWseWJM2Qt/5LUiPWPNCT/HqSe5L8KMmKn/yuNMxAkk1JDiQ5nOTWJGdPsLahx05yYZJDS36+leS67rF3JvnGkseumlVd3XYPdtscSnJw1P2nVVuSC5Lc3v09H0jy9iWPTfScDRueIoven+TeJF9cerfztIe26FHbG7vzeCTJnUv/faz0t51hbZcleWLJ3+lP+u475bp+f0lNR5L8MMmm7rFpn7OPJDmW5MgKj0//tVZVa/oD/BxwIXAbMLfCNk8HHmTxMsnTgYPAC7rH/hL4vW7+euD9E6xtpGMDpwGPAM/rlt8JvG0K56xXXd05O3e1v9ekawPOAy7p5s8CvgJsm/Q5O9nrZsk2rwU+CQR4AfClvvvOoLYXAc/u5q8EDg37286wtsuAT42z7zTrOmH7VwGfn8U5647/K91r6MgKj0/9tbbmLfSquq+qht1sdLJhBrYDe7r5vUx2+IFRj/1y4KtV9bUJ1rCc1f7Oa3rOquqRqrq7m/82cDfw3AnWcFyf4Sm2s3iFVlXVXcBTk1zQc9+p1lZVX6iqJ7rFO5jOORqrtintO+ljvx64eULPPVRV3Q588ySbTP21tuaB3tPJhhkYVNUCQDfdPMHnHfXYr+PJL6DfTnJ/ko8mOWfGdRVwvPvjrWPsP83aAEiyFbiUxcA6blLnrM/wFCttM+2hLUY9/puBf1yyvNLfdpa1vbjrJvh8km0j7jvNukhyBnAF8LElq6d5zvqY+mttJt8pmuRzLL7NPtEfVdUnZ1HDSk5W24jHeRrwauCGJav/GvhTFl9I7wTeD7xhhnW9uKoeSbIZ+EyS+6vqwAj7T7M2kjwTuAW4bklLdOxz1qoklwFvAn5pyeqp/G1HcCdwQVV9N8mvAp9I8vwZPv8wrwL+paqWtpjX+pxN3UwCvaouX+UhTjbMwEKSQVUtJBkAxyZVW5JRjn0lcFdVHV1y7IUlx/oAi58TzKyuqnqkmx5LcguLLeEDrINzlsW7iz8G3FxVH19y7LHP2TL6DE9xfJt/O2Gb03vsuxq9hs5IcgnwYeDKqnrs+PqT/G1nUlvXVXZ8/rNJ/pfF/+R7/V7TqmuJJ71bnvI562Pqr7WN0uVysmEG9gM7uvkdTHb4gVGO/aT+uq4lcNxrgXtnVVeSM7u3nSQ5k8W3n/f23X/KtYXFkLqvqt59wmOTPGd9hqfYT/cOoLvq4EdV9VDPfVdj6PGTbAE+Dryxqh5Ysv5kf9tZ1TZYMv9C4Jks/uc9zfPW69hJng28lMUPII+vm/Y562P6r7VpfeLb9wd4DYv/G30POAp8tlv/HGD/ku2uAu4B7mOxq+b4+nOAzwGHu+mmCda27LGXqe1M4DG6KxKWrN/L4gd+9wO3svgWdSZ1Ac/vnvtLLF5F8mf8+EayNT1nLHYdVFffoe7nqmmcs+VeN8BbgLd082Gxm+fero65k+074df+sNo+BDy+5BwdHPa3nWFt1wJHup+7gMtmcd6G1dUt/wbwdyfsN4tzdjPwMPB9FjPtTbN+rXmnqCQ1YqN0uUiShjDQJakRBrokNcJAl6RGGOiS1AgDXZIaYaBLUiMMdElqxP8BfwCDi9wrm3IAAAAASUVORK5CYII=\n",
"text/plain": [
"<Figure size 432x288 with 1 Axes>"
]
},
"metadata": {
"needs_background": "light"
},
"output_type": "display_data"
}
],
"source": [
"plt.hist(np.concatenate([MC_Dplus_tuple_dict_sig_for_BDT['cos_thetal'],MC_Ds_tuple_dict_sig_for_BDT['cos_thetal']]), bins=70);"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 2",
"language": "python",
"name": "python2"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 2
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython2",
"version": "2.7.15"
}
},
"nbformat": 4,
"nbformat_minor": 2
}
Davide Lancierini