{ "cells": [ { "cell_type": "code", "execution_count": 60, "metadata": {}, "outputs": [], "source": [ "import uproot\n", "import numpy as np\n", "import sys\n", "import os\n", "import matplotlib\n", "import matplotlib.pyplot as plt\n", "from mpl_toolkits import mplot3d\n", "import itertools\n", "import awkward as ak\n", "from scipy.optimize import curve_fit\n", "%matplotlib inline" ] }, { "cell_type": "code", "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": [ "