{ "cells": [ { "cell_type": "code", "execution_count": 7, "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", "import seaborn as sns\n", "%matplotlib inline" ] }, { "cell_type": "code", "execution_count": 31, "metadata": {}, "outputs": [], "source": [ "file = uproot.open(\"tracking_losses_ntuple_Bd2KstEE.root:PrDebugTrackingLosses.PrDebugTrackingTool/Tuple;1\")\n", "#file = uproot.open(\"tracking_losses_ntuple_Dst0ToD0EE.root:PrDebugTrackingLosses.PrDebugTrackingTool/Tuple;1\")\n", "\n", "#look at particles only from Signal\n", "allcolumns = file.arrays()\n", "tracked = allcolumns[(allcolumns.isElectron) & (~allcolumns.lost) & (allcolumns.fromSignal)]\n", "lost = allcolumns[(allcolumns.isElectron) & (allcolumns.lost) & (allcolumns.fromSignal)] \n", "\n", "#ak.num(tracked, axis=0)\n", "\n" ] }, { "cell_type": "code", "execution_count": 143, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
[[-676, -685, -694, -703, -764, -773, ..., -792, -854, -864, -864, -873, -882],\n", " [-1.4e+03, -1.45e+03, -1.5e+03, -1.54e+03, ..., -2.4e+03, -2.45e+03, -2.5e+03],\n", " [-148, -148, -149, -150, -156, -156, ..., -157, -158, -164, -164, -165, -166],\n", " [21.7, 23, 24.2, 25.4, 34, 35.3, 36.6, 37.9, 46.9, 48.2, 49.6, 50.9],\n", " [249, 250, 250, 250, 252, 252, 252, 252, 253, 253, 253, 254],\n", " [-134, -139, -145, -151, -191, -197, -203, -209, -251, -258, -264, -270],\n", " [213, 217, 220, 223, 244, 247, 250, 253, 275, 278, 282, 285],\n", " [1.17e+03, 1.19e+03, 1.21e+03, 1.23e+03, ..., 1.53e+03, 1.55e+03, 1.57e+03],\n", " [-135, -142, -149, -156, -207, -214, ..., -229, -229, -281, -289, -297, -304],\n", " [456, 460, 465, 469, 500, 504, 509, 514, 544, 549, 553, 558, 558],\n", " ...,\n", " [616, 629, 642, 655, 745, 758, 772, 785, 785, 878, 891, 905, 919],\n", " [-266, -267, -268, -269, -277, -277, ..., -279, -280, -287, -288, -289, -290],\n", " [-76.4, -73.2, -69.9, -66.6, -43.8, ..., -36.8, -33.3, -8.97, -5.34, -1.68],\n", " [-92, -90.5, -88.9, -87.3, -75.7, ..., -70.3, -57.5, -55.5, -53.6, -51.6],\n", " [-518, -527, -536, -545, -608, -617, -627, -636, -700, -710, -719, -729],\n", " [510, 517, 523, 576, 583, 583, 590, 597, 643, 650, 657],\n", " [1.2e+03, 1.21e+03, 1.21e+03, 1.22e+03, ..., 1.35e+03, 1.36e+03, 1.37e+03],\n", " [-1.33e+03, -1.33e+03, -1.34e+03, ..., -1.39e+03, -1.39e+03, -1.4e+03],\n", " [-1.27e+03, -1.29e+03, -1.32e+03, ..., -1.72e+03, -1.74e+03, -1.76e+03]]\n", "--------------------------------------------------------------------------------\n", "type: 9489 * var * float32" ], "text/plain": [ "