cetin/Projektpraktikum
1
1
Fork 0
Code Issues Pull Requests repo.project_board Releases Wiki Activity
b414440cb5
Projektpraktikum/notebooks/trackinglosses_photons.ipynb

540 lines
55 KiB
Plaintext
Raw Normal View History

rad length
2024-01-19 11:22:15 +01:00
{
"cells": [
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 1,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [],
"source": [
"import uproot\t\n",
"import numpy as np\n",
"import matplotlib.pyplot as plt\n",
"from mpl_toolkits import mplot3d\n",
"import awkward as ak\n",
"from scipy.optimize import curve_fit\n",
thesis
2024-02-04 16:56:05 +01:00
"import mplhep\n",
"mplhep.style.use([\"LHCbTex2\"])\n",
"\n",
"plt.rcParams[\"savefig.dpi\"] = 600\n",
rad length
2024-01-19 11:22:15 +01:00
"%matplotlib inline"
]
},
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 2,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [],
"source": [
thesis
2024-02-05 15:44:27 +01:00
"file = uproot.open(\n",
" \"/work/cetin/Projektpraktikum/trackinglosses_B_photon_cuts.root\")\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
"# selektiere nur elektronen von B->K*ee\n",
"allcolumns = []\n",
"for i in range(11):\n",
thesis
2024-02-04 16:56:05 +01:00
" allcolumns.append(file[\"Tree\" + str(i)].arrays())"
rad length
2024-01-19 11:22:15 +01:00
]
},
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 3,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<pre>{oneCut_event_id: 1,\n",
" oneCut_lost: False,\n",
" oneCut_rad_length_frac: 0.148,\n",
" oneCut_energy: 1.28e+04,\n",
" noneCut_brem_photons_pe: 1,\n",
" oneCut_brem_photons_pe: [7.42e+03],\n",
" noneCut_brem_vtx_x: 1,\n",
" oneCut_brem_vtx_x: [-3.61],\n",
" noneCut_brem_vtx_z: 1,\n",
" oneCut_brem_vtx_z: [35.6],\n",
" oneCut_photon_length: 1}\n",
"------------------------------------------\n",
"type: {\n",
" oneCut_event_id: int64,\n",
" oneCut_lost: bool,\n",
" oneCut_rad_length_frac: float64,\n",
" oneCut_energy: float64,\n",
" noneCut_brem_photons_pe: int32,\n",
" oneCut_brem_photons_pe: var * float64,\n",
" noneCut_brem_vtx_x: int32,\n",
" oneCut_brem_vtx_x: var * float64,\n",
" noneCut_brem_vtx_z: int32,\n",
" oneCut_brem_vtx_z: var * float64,\n",
" oneCut_photon_length: int64\n",
"}</pre>"
],
"text/plain": [
"<Record {oneCut_event_id: 1, ...} type='{oneCut_event_id: int64, oneCut_los...'>"
]
},
thur
2024-02-08 17:43:22 +01:00
"execution_count": 3,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"allcolumns[1][1]"
]
},
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 4,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [],
"source": [
"def cutdict():\n",
" basedict = {\n",
thesis
2024-02-04 16:56:05 +01:00
" \"0\": {},\n",
" \"1\": {},\n",
" \"2\": {},\n",
" \"3\": {},\n",
" \"4\": {},\n",
" \"5\": {},\n",
" \"6\": {},\n",
" \"7\": {},\n",
" \"8\": {},\n",
" \"9\": {},\n",
" \"10\": {},\n",
" }\n",
"\n",
rad length
2024-01-19 11:22:15 +01:00
" basedict[\"0\"] = \"no\"\n",
" basedict[\"1\"] = \"one\"\n",
" basedict[\"2\"] = \"two\"\n",
" basedict[\"3\"] = \"three\"\n",
" basedict[\"4\"] = \"four\"\n",
" basedict[\"5\"] = \"five\"\n",
" basedict[\"6\"] = \"six\"\n",
" basedict[\"7\"] = \"seven\"\n",
" basedict[\"8\"] = \"eight\"\n",
" basedict[\"9\"] = \"nine\"\n",
" basedict[\"10\"] = \"ten\"\n",
thesis
2024-02-04 16:56:05 +01:00
"\n",
rad length
2024-01-19 11:22:15 +01:00
" return basedict\n",
"\n",
thesis
2024-02-04 16:56:05 +01:00
"\n",
rad length
2024-01-19 11:22:15 +01:00
"Cuts = cutdict()"
]
},
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 5,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [],
"source": [
thesis
2024-02-04 16:56:05 +01:00
"# electrons = []\n",
"# for jcut in range(11):\n",
"\n",
"jcut = 4 # cut 0.2*E\n",
"\n",
"energy_emissions = ak.ArrayBuilder()\n",
"\n",
"for jelec in range(ak.num(allcolumns[jcut], axis=0)):\n",
" energy_emissions.begin_record()\n",
" energy_emissions.field(\"lost\").boolean(\n",
thesis
2024-02-05 15:44:27 +01:00
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"lost\"]\n",
" )\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"rad_length_frac\").real(\n",
thesis
2024-02-05 15:44:27 +01:00
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"rad_length_frac\"]\n",
" )\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"energy\").real(\n",
thesis
2024-02-05 15:44:27 +01:00
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"energy\"]\n",
" )\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
" tmp_velo = 0\n",
" tmp_richut = 0\n",
" tmp_neither = 0\n",
" tmp_velo_length = 0\n",
" tmp_richut_length = 0\n",
" tmp_neither_length = 0\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
" for jphoton in range(\n",
thesis
2024-02-05 15:44:27 +01:00
" ak.num(\n",
" allcolumns[jcut][jelec][Cuts[str(jcut)] + \"Cut_\" + \"brem_photons_pe\"],\n",
" axis=0,\n",
" )\n",
" ):\n",
" if (\n",
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_vtx_z\", jphoton]\n",
" <= 770\n",
" ):\n",
" tmp_velo += allcolumns[jcut][\n",
" jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_photons_pe\", jphoton\n",
" ]\n",
thesis
2024-02-04 16:56:05 +01:00
" tmp_velo_length += 1\n",
thesis
2024-02-05 15:44:27 +01:00
" elif (\n",
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_vtx_z\", jphoton]\n",
" > 770\n",
" ) and (\n",
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_vtx_z\", jphoton]\n",
" <= 2700\n",
" ):\n",
" tmp_richut += allcolumns[jcut][\n",
" jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_photons_pe\", jphoton\n",
" ]\n",
thesis
2024-02-04 16:56:05 +01:00
" tmp_richut_length += 1\n",
" else:\n",
thesis
2024-02-05 15:44:27 +01:00
" tmp_neither += allcolumns[jcut][\n",
" jelec, Cuts[str(jcut)] + \"Cut_\" + \"brem_photons_pe\", jphoton\n",
" ]\n",
thesis
2024-02-04 16:56:05 +01:00
" tmp_neither_length += 1\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"velo_length\").integer(tmp_velo_length)\n",
" energy_emissions.field(\"velo\").real(tmp_velo)\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"rich_length\").integer(tmp_richut_length)\n",
" energy_emissions.field(\"rich\").real(tmp_richut)\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"neither_length\").integer(tmp_neither_length)\n",
" energy_emissions.field(\"downstream\").real(tmp_neither)\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-05 15:44:27 +01:00
" energy_emissions.field(\"photon_length\").integer(tmp_richut_length + tmp_velo_length)\n",
thesis
2024-02-04 16:56:05 +01:00
"\n",
thesis
2024-02-05 15:44:27 +01:00
" if (\n",
" (tmp_velo == 0)\n",
" and (tmp_richut == 0)\n",
" or (\n",
" allcolumns[jcut][jelec, Cuts[str(jcut)] + \"Cut_\" + \"energy\"] - tmp_velo\n",
" < 3000\n",
" )\n",
" ):\n",
thesis
2024-02-04 16:56:05 +01:00
" energy_emissions.field(\"quality\").integer(0)\n",
" else:\n",
" energy_emissions.field(\"quality\").integer(1)\n",
"\n",
" energy_emissions.end_record()\n",
"\n",
"energy_emissions = ak.Array(energy_emissions)\n",
"# electrons.append(energy_emissions)"
rad length
2024-01-19 11:22:15 +01:00
]
},
{
"cell_type": "code",
thur
2024-02-08 17:43:22 +01:00
"execution_count": 6,
rad length
2024-01-19 11:22:15 +01:00
"metadata": {},
"outputs": [
{
"data": {
thur
2024-02-08 17:43:22 +01:00
"image/png": "iVBORw0KGgoAAAANSUhEUgAABMEAAAOWCAYAAADm+IXMAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAAB2uElEQVR4nOz9z29jaZon9j6KThjVsFtNRQEGhNBignLPphfdQ0U2UEs7SANem8xY9sIIcoBaXECwxY75AzpBDlrAXZQxZPSmcBfXkdTYuCsPSoy2gbsowBmipxa1ssXqRQR0YaBCLM0MKlEoi3eRTTZJURJ/iT+OPh9ACFE6fPkc8hwp+NXzvmer2+12AwAAAAAS7MmqCwAAAACAhyYEAwAAACDxhGAAAAAAJJ4QDAAAAIDEE4IBAAAAkHhCMAAAAAASTwgGAAAAQOIJwQAAAABIPCEYAAAAAIknBAMAAAAg8YRgAAAAACSeEAwAAACAxBOCAQAAAJB4QjAAAAAAEk8IBgAAAEDiCcEAAAAASDwhGAAAAACJJwQDAAAAIPGEYAAAAAAknhAMAAAAgMQTggEAAACQeEIwAAAAABJPCAZT6nQ6cXJysuoyAAAAgCkIwWCMVqsVW1tbYz92dnYilUqtukQAAABgCkIwGOPrr7++9XvpdDqy2ewSqwEAAADmtdXtdrurLgLWSbvdjv39/Wg0GmM7vtLpdKTT6eUXBgAAAMxMCAYjSqVSfP78ORqNxqpLAQAAABbki1UXAOuk0+lEvV4XgAEAAEDCWBMMBvTWAisUCrG1tRUHBwdRrVaj0+mstjAAAABgLkIw5tJutyOXy8XJyclc41Sr1cjlcrGzsxNbW1uxv78fhUIhms3mgiqdvI5BrVYryuVy7OzsRLlcXmotAAAAwOIIwZhJp9OJQqEQ+/v70Ww24/PnzzON02w2hwKmRqMR5+fnUalUotVqRS6Xi1wut7ROrMvLyzg7O4tarRb5fH7oe72gDgAAANg8FsZnKp1OJ77++usbHVO1Wi2KxeJUYzWbzX6oVCwWo1ar3djm4OAgWq1WpNPpODs7G3u1xoc0bn8rlUocHR0ttQ4AAABgPjrBmFi1Wu2HUvPqdZJFRKTT6bEBWET0F6hvt9v97ZcplUpFpVIZCuBMiwQAAIDNIwRjIq1WK7LZbJyfn8fp6emtodWkCoVCf4rjXaFSOp3uT0tsNptRr9fnetxZZTKZqFQq/dvLXqsMAAAAmI8QjIlkMpnIZDL92y9evJh5rHa7PRQiffXVV3du/+rVq/7n4wKzVqsVW1tbM31M011WLBb73WCuFgkAAACb5YtVF8Ds2u12RHzfLTWvZrMZ2Wx24u3nWZtrsKMqm83eO9bgAvWdTidOTk6GvpZKpW4sYj+paRe6f/HiRTSbzaWvTQYAAADMRwi2odrtdhwcHERExNnZ2VxBWKFQiJOTk6Ut+D44pXGwu+wu6XS6H/q9e/duKPRKp9P9tcOWZZ5OOAAAAGD5TIfcUO12OzqdTnQ6nTg4OOgHRNPqBWAREd9+++0iSxxrdFH9L7/8cqL7DYZlvXpXod1uRyqV0gkGAAAAG0YItqGy2Wy/+2nWIGwwAMvn80vpphpdUH7SDrbR7RZxhcpptVqtaLfb8ebNm6U/NgAAADAfIdgGGwyupg3CVhGARdzsNpu0o+qHP/zh0O0PHz4sqqS+crkcuVzu1k6z169fRz6fX8qUUQAAAGCxhGAbbpYgbFUBWETcqG3WTrDz8/OF1dRTrVaj2WxGoVCIg4ODaDab/YX49/f3h7rvAAAAgM0iBEuAaYKwVQZgETdDsFl1Op2FjDPo7Oysf7XKVqsVhUIhXr58Gd9++22cnp4OXdUSAAAA2CyuDpkQvUCrUCj0g7DRq0auOgCLmD28Gp02+fnz5/mLGZHJZOL09HTh4wIAAACrpxMsQe7qCCuVSisPwBbpITrBAAAAgOQSgiXMaBCWy+WiUChEvV6/8f1VmHQh/GWNAwAAADwOQrAEGgy62u32WnWAPX36dK3GAQAAAB4HIVhC5fP5yGazQ197+/btiqr5R7N2cI1Of9QJBgAAAExDCJZQhUIhms3m0NcODg5WvpbWixcvhm5PWs/oQvj7+/uLKgkAAAB4BIRgCXTbVSDb7fbKg7CDg4Oh272F++9zfn4+dHu0yw0AAADgLkKwhBkXgK1TEDbaCTZpCDZYbyqVinQ6vciyAAAAgIQTgiXIbR1go7dXGYRlMpmh9by+/fbbie734cOH/uejQRoAAADAfYRgCVEqle69CuS6BGFfffVV//NWqzXRfQa3K5fLC68JAAAASDYhWAKUSqWo1+sRcXsA1rMOQVipVOp/Prp4/ziD26TTaeuBAQAAAFMTgm24aQKwnkUEYfMEZ5lMZijI6nWw3WZwn3SBAQAAALMQgm2wWQKwnnmDsNEF7acNxWq1Wv/zr7/++tbtOp1Ofx+z2WwUi8WpHgcAAAAgQgi2ser1+swBWM9oEPby5cuJ7tfpdG50ZL17926qx06n0/3HbrVaUa1Wx27XqymVSs20jwAAAAAREVvdbre76iKYTS6XW0g4dHJyEq9fv46zs7NIp9Njt+l0OvH69evodDp3ruOVzWYjlUrFmzdvIpPJ3PvYzWYzCoVCdDqdyOfz8ebNm0in0/Hhw4col8vRarUik8nE+/fvh64qCQAAADANIRhroVqtxrt376Ldbken04lUKhUvXryIUqkU+Xx+1eUBAAAAG04IBgAAAEDiWRMMAAAAgMT7YtUFML3/+D/+j+O7776LP/iDP4j/9D/9T1ddDgAAAMCd/q//6/+K//v//r/jBz/4QfyH//AfVlKD6ZAb6MmTJ+FlAwAAADbN1tZWXF9fr+SxdYJtoMEA7NmzZyusZHYXFxdxfX0dT548id3d3VWXM7VNrz/CPqwL+7Ae7MN62PR92PT6I+zDurAP68E+rIdN34dNrz/CPqyLJOzDp0+fIiJW2tQjBNtAT5486R/8Hz9+XHU5M9nb24tPnz7F7u7uRu7DptcfYR/WhX1YD/ZhPWz6Pmx6/RH2YV3Yh/VgH9bDpu/DptcfYR/WRRL24Q/+4A/6WcaqWBgfAAAAgMRLdAj2m9/8Jl69ehV/8Ad/EH/xF38RV1dXqy4JAAAAgBVIdAhWLpej0WhEt9uNs7Oz+Oqrr1ZdEgAAAAArkOg1wZrNZmxtbUXE9wuvnZ6errgiAAAAAFYh0Z1gL1++HLrqQDabXWE1AAAAAKxKojvBarVaRES8f/8+MplMvH37dsUVAQAAALAKiQ7BIv4xCAMAAADg8Up8CJZEf/RHfxS/+c1v4o/+6I9WXcqjdXh4GFdXV7G9vb3qUmZmH9ZDEvYhCZLwOtiH1dv0+pMiCa+DfVgPSdiHJNj012HT64+wDyzOOmQZW93BRbPYCHt7e/Hp06d49uxZfPz4cdXlzCQJ+wB8z/kMyeF8hmRxTkNyJOF8Xod9eBSdYH//938f/+Sf/JNVl7FwFxcXsbe3d+c2h4eHcXh4uKSKAAAAgMfm+Pg4jo+P79zm4uJiSdXcLvEh2L/+1/86vvrqqzg7O4s///M/X3U5C3V9fR2fPn26c5urq6slVQMAAAA8RldXV/fmE+sg8SHYt99+G91uN7755pvEhWBPnjyJ3d3dO7cx5xkAAAB4SNvb2/Hs2bM7t7m4uIjr6+slVTTeWodgf/u3fzvzfTudTpyfn0e9Xo+tra04PT2Nv/7rv15gdau3u7u7sXOBAQAAgGSYZCmm3ppgq7TWIVixWIytra25xuit+99qtRK7NhgAAAAAd3uy6gImMcsFLLvdbnS73dja2uoHafV6fdGlAQAAALABNiIEi/jHUGvSj4iIra2toa81Go0V7wUAAAAAq7DW0yEjIp4/fx7lcjnS6fRU9zs/P49qtRoHBwfx9u1bC8SvmcPDw7i6uvK6QAI4nyE5nM+QLM5pSA7n82JsdWeZa7gkT548iXa7Pdc6Xv/Zf/afxYsXL+J/+B/+h8UVtmK9xeSePXtmYXwAAABg7a1DlrHW0yFTqdTcC9n/q3/1r+Kbb76Jv/mbv1l
rad length
2024-01-19 11:22:15 +01:00
"text/plain": [
thesis
2024-02-04 16:56:05 +01:00
"<Figure size 1200x900 with 1 Axes>"
rad length
2024-01-19 11:22:15 +01:00
]
},
"metadata": {},
"output_type": "display_data"
}
],
thesis
2024-02-04 16:56:05 +01:00
"source": [
"velo_found = ak.to_numpy(\n",
" energy_emissions[(~energy_emissions.lost) & (energy_emissions.quality == 1)][\"velo\"]\n",
")\n",
"rich_found = ak.to_numpy(\n",
" energy_emissions[(~energy_emissions.lost) & (energy_emissions.quality == 1)][\"rich\"]\n",
")\n",
"energy_found = ak.to_numpy(\n",
" energy_emissions[(~energy_emissions.lost) & (energy_emissions.quality == 1)][\n",
" \"energy\"\n",
" ]\n",
")\n",
"\n",
"velo_lost = ak.to_numpy(\n",
" energy_emissions[(energy_emissions.lost) & (energy_emissions.quality == 1)][\"velo\"]\n",
")\n",
"rich_lost = ak.to_numpy(\n",
" energy_emissions[(energy_emissions.lost) & (energy_emissions.quality == 1)][\"rich\"]\n",
")\n",
"energy_lost = ak.to_numpy(\n",
" energy_emissions[(energy_emissions.lost) & (energy_emissions.quality == 1)][\n",
" \"energy\"\n",
" ]\n",
")\n",
"\n",
"diff_found = velo_found - rich_found\n",
"diff_lost = velo_lost - rich_lost\n",
"\n",
"xlim = 20000\n",
"nbins = 60\n",
"\n",
"plt.hist(\n",
" diff_lost,\n",
" bins=nbins,\n",
" density=True,\n",
" alpha=0.5,\n",
" histtype=\"bar\",\n",
" color=\"#F05342\",\n",
" label=\"lost\",\n",
" range=[-xlim, xlim],\n",
")\n",
"plt.hist(\n",
" diff_found,\n",
" bins=nbins,\n",
" density=True,\n",
" alpha=0.5,\n",
" histtype=\"bar\",\n",
thesis
2024-02-05 15:44:27 +01:00
" color=\"#2A9D8F\", # \"#107E7D\",\n",
thesis
2024-02-04 16:56:05 +01:00
" label=\"found\",\n",
" range=[-xlim, xlim],\n",
")\n",
"# plt.xlim(-20000, 20000)\n",
"# plt.yscale(\"log\")\n",
"# plt.title(\"emitted energy difference\")\n",
thesis
2024-02-05 15:44:27 +01:00
"plt.xlabel(r\"$\\Delta E_{VELO} - \\Delta E_{RICH1+UT}$ [MeV]\")\n",
thesis
2024-02-04 16:56:05 +01:00
"plt.ylabel(\"a.u.\")\n",
"plt.legend(title=\"LHCb Simulation\")\n",
thur
2024-02-08 17:43:22 +01:00
"# plt.show()\n",
thesis
2024-02-05 15:44:27 +01:00
"plt.savefig(\n",
" \"/work/cetin/Projektpraktikum/thesis/emitted_energy_difference.pdf\", format=\"PDF\"\n",
")"
thesis
2024-02-04 16:56:05 +01:00
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "markdown",
"metadata": {},
rad length
2024-01-19 11:22:15 +01:00
"source": [
"nbins = 6\n",
thesis
2024-02-04 16:56:05 +01:00
"quality_cut = electrons[jcut].quality != -1\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
"### all split in velo and rich\n",
"\n",
thesis
2024-02-04 16:56:05 +01:00
"fig, axs = plt.subplots(3, 3, figsize=(15, 6))\n",
rad length
2024-01-19 11:22:15 +01:00
"ax = axs.ravel()\n",
thesis
2024-02-04 16:56:05 +01:00
"for jcut, ax in enumerate(ax):\n",
"ax.hist(\n",
"ak.to_numpy(electrons[jcut][quality_cut][\"velo_length\"]),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"darkorange\",\n",
"histtype=\"bar\",\n",
"label=\"velo\",\n",
"range=[0, nbins],\n",
")\n",
"ax.hist(\n",
"ak.to_numpy(electrons[jcut][quality_cut][\"rich_length\"]),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"blue\",\n",
"histtype=\"bar\",\n",
"label=\"rich\",\n",
"range=[0, nbins],\n",
")\n",
"ax.set_xlim(0, nbins)\n",
"ax.set_ylim(0, 1)\n",
"ax.set_title(\"Photon Cut: \" + str(np.round(jcut \\* 0.05, 2)) + f\"$E_0$\")\n",
"ax.set_xlabel(\"number of photons\")\n",
"ax.set_ylabel(\"a.u.\")\n",
rad length
2024-01-19 11:22:15 +01:00
"plt.suptitle(\"number of photons in velo and rich\")\n",
"plt.legend()\n",
"plt.tight_layout()\n",
"plt.show()\n",
"\n",
"### found\n",
"\n",
thesis
2024-02-04 16:56:05 +01:00
"fig, axs = plt.subplots(3, 3, figsize=(15, 6))\n",
rad length
2024-01-19 11:22:15 +01:00
"ax = axs.ravel()\n",
thesis
2024-02-04 16:56:05 +01:00
"for jcut, ax in enumerate(ax):\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut]~(electrons[jcut].lost) & quality_cut][\"velo_length\"]\n",
"),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"darkorange\",\n",
"histtype=\"bar\",\n",
"label=\"velo\",\n",
"range=[0, nbins],\n",
")\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut]~(electrons[jcut].lost) & quality_cut][\"rich_length\"]\n",
"),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"blue\",\n",
"histtype=\"bar\",\n",
"label=\"rich\",\n",
"range=[0, nbins],\n",
")\n",
"ax.set_xlim(0, nbins)\n",
"ax.set_ylim(0, 1)\n",
"ax.set_title(\"Photon Cut: \" + str(np.round(jcut \\* 0.05, 2)) + f\"$E_0$\")\n",
"ax.set_xlabel(\"number of photons\")\n",
"ax.set_ylabel(\"a.u.\")\n",
rad length
2024-01-19 11:22:15 +01:00
"plt.suptitle(\"number of photons of found electrons\")\n",
"plt.legend()\n",
"plt.tight_layout()\n",
"plt.show()\n",
"\n",
thesis
2024-02-04 16:56:05 +01:00
"### lost\n",
rad length
2024-01-19 11:22:15 +01:00
"\n",
thesis
2024-02-04 16:56:05 +01:00
"fig, axs = plt.subplots(3, 3, figsize=(15, 6))\n",
rad length
2024-01-19 11:22:15 +01:00
"ax = axs.ravel()\n",
thesis
2024-02-04 16:56:05 +01:00
"for jcut, ax in enumerate(ax):\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut](electrons[jcut].lost) & quality_cut][\"velo_length\"]\n",
"),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"darkorange\",\n",
"histtype=\"bar\",\n",
"label=\"velo\",\n",
"range=[0, nbins],\n",
")\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut](electrons[jcut].lost) & quality_cut][\"rich_length\"]\n",
"),\n",
"bins=nbins,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"blue\",\n",
"histtype=\"bar\",\n",
"label=\"rich\",\n",
"range=[0, nbins],\n",
")\n",
"ax.set_xlim(0, nbins)\n",
"ax.set_ylim(0, 1)\n",
"ax.set_title(\"Photon Cut: \" + str(np.round(jcut \\* 0.05, 2)) + f\"$E_0$\")\n",
"ax.set_xlabel(\"number of photons\")\n",
"ax.set_ylabel(\"a.u.\")\n",
rad length
2024-01-19 11:22:15 +01:00
"plt.suptitle(\"number of photons of lost electrons\")\n",
"plt.legend()\n",
"plt.tight_layout()\n",
thesis
2024-02-04 16:56:05 +01:00
"plt.show()\n",
rad length
2024-01-19 11:22:15 +01:00
"quality_cut = electrons[jcut].quality != -1\n",
thesis
2024-02-04 16:56:05 +01:00
"\n",
rad length
2024-01-19 11:22:15 +01:00
"### all split in lost and found\n",
"\n",
thesis
2024-02-04 16:56:05 +01:00
"fig, axs = plt.subplots(3, 3, figsize=(15, 6))\n",
rad length
2024-01-19 11:22:15 +01:00
"ax = axs.ravel()\n",
thesis
2024-02-04 16:56:05 +01:00
"for jcut, ax in enumerate(ax):\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut](electrons[jcut].lost) & (quality_cut)][\"photon_length\"]\n",
"),\n",
"bins=10,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"darkorange\",\n",
"histtype=\"bar\",\n",
"label=\"lost\",\n",
"range=[0, 10],\n",
")\n",
"ax.hist(\n",
"ak.to_numpy(\n",
"electrons[jcut](~electrons[jcut].lost) & (quality_cut)][\"photon_length\"]\n",
"),\n",
"bins=10,\n",
"density=True,\n",
"alpha=0.5,\n",
"color=\"blue\",\n",
"histtype=\"bar\",\n",
"label=\"found\",\n",
"range=[0, 10],\n",
")\n",
"ax.set_xlim(0, 10) # ax.set_ylim(0,1) # ax.set_yscale('log')\n",
"ax.set_title(\"Photon Cut: \" + str(np.round(jcut \\* 0.05, 2)) + f\"$E_0$\")\n",
"ax.set_xlabel(\"number of photons\")\n",
"ax.set_ylabel(\"a.u.\")\n",
rad length
2024-01-19 11:22:15 +01:00
"plt.suptitle(\"number of photons in lost and found\")\n",
"plt.legend()\n",
"plt.tight_layout()\n",
"plt.show()\n"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
thesis
2024-02-04 16:56:05 +01:00
"source": []
rad length
2024-01-19 11:22:15 +01:00
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "tuner",
"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.10.12"
}
},
"nbformat": 4,
"nbformat_minor": 2
}
Reference in New Issue Copy Permalink