diff --git a/D_tasks.ipynb b/D_tasks.ipynb index cd918a0..a08be4c 100644 --- a/D_tasks.ipynb +++ b/D_tasks.ipynb @@ -21,16 +21,128 @@ }, { "cell_type": "code", - "execution_count": 61, + "execution_count": 86, "metadata": {}, "outputs": [ { "data": { + "text/html": [ + "
{all_endvtx_types_length: 5,\n",
+       " all_endvtx_types: [101, 101, 101, 101, 0],\n",
+       " all_endvtx_x_length: 5,\n",
+       " all_endvtx_x: [-3.52e+03, -3.52e+03, -3.52e+03, -3.53e+03, -3.54e+03],\n",
+       " all_endvtx_y_length: 5,\n",
+       " all_endvtx_y: [-302, -302, -302, -302, -297],\n",
+       " all_endvtx_z_length: 5,\n",
+       " all_endvtx_z: [9.84e+03, 9.84e+03, 9.85e+03, 9.85e+03, 9.87e+03],\n",
+       " brem_photons_pe_length: 4,\n",
+       " brem_photons_pe: [519, 484, 173, 466],\n",
+       " brem_photons_px_length: 4,\n",
+       " brem_photons_px: [-327, -306, -109, -283],\n",
+       " brem_photons_py_length: 4,\n",
+       " brem_photons_py: [-8.03, -7.15, -4.38, -13.6],\n",
+       " brem_photons_pz_length: 4,\n",
+       " brem_photons_pz: [403, 374, 135, 370],\n",
+       " brem_vtx_x_length: 4,\n",
+       " brem_vtx_x: [-3.52e+03, -3.52e+03, -3.52e+03, -3.53e+03],\n",
+       " brem_vtx_y_length: 4,\n",
+       " ...}\n",
+       "-----------------------------------------------------------------------\n",
+       "type: {\n",
+       "    all_endvtx_types_length: int32,\n",
+       "    all_endvtx_types: var * float32,\n",
+       "    all_endvtx_x_length: int32,\n",
+       "    all_endvtx_x: var * float32,\n",
+       "    all_endvtx_y_length: int32,\n",
+       "    all_endvtx_y: var * float32,\n",
+       "    all_endvtx_z_length: int32,\n",
+       "    all_endvtx_z: var * float32,\n",
+       "    brem_photons_pe_length: int32,\n",
+       "    brem_photons_pe: var * float32,\n",
+       "    brem_photons_px_length: int32,\n",
+       "    brem_photons_px: var * float32,\n",
+       "    brem_photons_py_length: int32,\n",
+       "    brem_photons_py: var * float32,\n",
+       "    brem_photons_pz_length: int32,\n",
+       "    brem_photons_pz: var * float32,\n",
+       "    brem_vtx_x_length: int32,\n",
+       "    brem_vtx_x: var * float32,\n",
+       "    brem_vtx_y_length: int32,\n",
+       "    brem_vtx_y: var * float32,\n",
+       "    brem_vtx_z_length: int32,\n",
+       "    brem_vtx_z: var * float32,\n",
+       "    endvtx_type: int32,\n",
+       "    endvtx_x: float64,\n",
+       "    endvtx_y: float64,\n",
+       "    endvtx_z: float64,\n",
+       "    energy: float64,\n",
+       "    eta: float64,\n",
+       "    event_count: int32,\n",
+       "    fromB: bool,\n",
+       "    fromD: bool,\n",
+       "    fromDecay: bool,\n",
+       "    fromHadInt: bool,\n",
+       "    fromPV: bool,\n",
+       "    fromPairProd: bool,\n",
+       "    fromSignal: bool,\n",
+       "    fromStrange: bool,\n",
+       "    isElectron: bool,\n",
+       "    isKaon: bool,\n",
+       "    isMuon: bool,\n",
+       "    isPion: bool,\n",
+       "    isProton: bool,\n",
+       "    lost: bool,\n",
+       "    lost_in_track_fit: bool,\n",
+       "    match_fraction: float32,\n",
+       "    mcp_idx: int32,\n",
+       "    mother_id: int32,\n",
+       "    mother_key: int32,\n",
+       "    originvtx_type: int32,\n",
+       "    originvtx_x: float64,\n",
+       "    originvtx_y: float64,\n",
+       "    originvtx_z: float64,\n",
+       "    p: float64,\n",
+       "    phi: float64,\n",
+       "    pid: int32,\n",
+       "    pt: float64,\n",
+       "    px: float64,\n",
+       "    py: float64,\n",
+       "    pz: float64,\n",
+       "    scifi_hit_pos_x_length: int32,\n",
+       "    scifi_hit_pos_x: var * float32,\n",
+       "    scifi_hit_pos_y_length: int32,\n",
+       "    scifi_hit_pos_y: var * float32,\n",
+       "    scifi_hit_pos_z_length: int32,\n",
+       "    scifi_hit_pos_z: var * float32,\n",
+       "    track_p: float64,\n",
+       "    track_pt: float64,\n",
+       "    tx: float64,\n",
+       "    ty: float64,\n",
+       "    ut_hit_pos_x_length: int32,\n",
+       "    ut_hit_pos_x: var * float32,\n",
+       "    ut_hit_pos_y_length: int32,\n",
+       "    ut_hit_pos_y: var * float32,\n",
+       "    ut_hit_pos_z_length: int32,\n",
+       "    ut_hit_pos_z: var * float32,\n",
+       "    velo_hit_pos_x_length: int32,\n",
+       "    velo_hit_pos_x: var * float32,\n",
+       "    velo_hit_pos_y_length: int32,\n",
+       "    velo_hit_pos_y: var * float32,\n",
+       "    velo_hit_pos_z_length: int32,\n",
+       "    velo_hit_pos_z: var * float32,\n",
+       "    velo_track_idx: int32,\n",
+       "    velo_track_tx: float64,\n",
+       "    velo_track_ty: float64,\n",
+       "    velo_track_x: float64,\n",
+       "    velo_track_y: float64,\n",
+       "    velo_track_z: float64\n",
+       "}
" + ], "text/plain": [ - "51" + "" ] }, - "execution_count": 61, + "execution_count": 86, "metadata": {}, "output_type": "execute_result" } @@ -45,7 +157,8 @@ "lost = allcolumns[(allcolumns.isElectron) & (allcolumns.lost) & (allcolumns.fromSignal) & (allcolumns.p < 5e3)] #D: 1908\n", "\n", "#ak.num(lost, axis=0)\n", - "ak.num(allcolumns[(allcolumns.fromPairProd) & (allcolumns.isElectron) & (~allcolumns.lost) & (allcolumns.fromSignal)],axis=0)" + "ak.num(allcolumns[(allcolumns.fromPairProd) & (allcolumns.isElectron) & (~allcolumns.lost) & (allcolumns.fromSignal)],axis=0)\n", + "found[0]" ] }, {