tracking-parametrisation-tuner/test/ghost_data_new_vars.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "import uproot\n",
    "import ROOT\n",
    "import matplotlib.pyplot as plt\n",
    "import awkward as ak\n",
    "import numpy as np\n",
    "from numpy import abs\n",
    "import math\n",
    "from array import array\n",
    "import mplhep\n",
    "mplhep.style.use([\"LHCbTex2\"])\n",
    "#%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "File = uproot.open(\"/work/cetin/LHCb/reco_tuner/data/ghost_data_B_new_vars_default_weights.root:PrMatchNN_3e224c41.PrMCDebugMatchToolNN/MVAInputAndOutput\")\n",
    "allcolumns = File.arrays()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "tracked = allcolumns[allcolumns[\"mc_quality\"]==-1]\n",
    "ghost = allcolumns[allcolumns[\"mc_quality\"]==0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "tracked_zmag = ak.to_numpy(tracked[\"mc_zMag\"])\n",
    "\n",
    "ghost_zmag = ak.to_numpy(ghost[\"mc_zMag\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABMIAAAOWCAYAAAANzz7PAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAAB6c0lEQVR4nOz9TWyjWZon9j6KSrf79njUVBQ8gCAtJsme3cAek5G9MdALB2ngbi5QMBm5qNqGiNnKY7HDgLcdQ41bS8Nk9LILcCZl1/L6jhizKOBuHEG6YdTOQ2ZfIAQBNirE0rhm7LZbvIswWaQ+KFLih6j39wOIJKXDcx4y3jcl/nXOeTf6/X4/AAAAAOCJe7bqAgAAAABgGQRhAAAAACSCIAwAAACARBCEAQAAAJAIgjAAAAAAEkEQBgAAAEAiCMIAAAAASARBGAAAAACJIAgDAAAAIBEEYQAAAAAkgiAMAAAAgEQQhAEAAACQCIIwAAAAABJBEAYAAABAIgjCAAAAAEgEQRgAAAAAiSAIAwAAACARBGEAAAAAJIIgDAAAAIBEEIQBAAAAkAiCMAAAAAASQRAGAAAAQCIIwgAAAABIBEEYAAAAAIkgCAMAAAAgEQRhAAAAACSCIAwAAACARBCEAQAAAJAIgjAAAAAAEkEQBgCMOT4+jkwmExsbGzfe6vX6g8eoVCq39r+1tRXNZnMOr4SH6na7cXh4GLlc7sH/7s1mM7a2tiKTySz133eeryHJFvXvt6rjAoDk2uj3+/1VFwEAPD69Xi8qlcq18CCdTken03lQ31tbW9Hr9ca+ls1m4/3795FKpR7UNw/TbDajVqtFs9kc+zeq1Wqxt7d3735H/83ncQxNsqjXkGSL+vdb5nEBABFmhAEAt0ilUlGr1a59vdvtxvHx8b37rdfr10KwiIh3794JwR6B58+fR7lcnntg9Pz587n2d9dYi3gNSbaof79lHhcAECEIAwDukE6nr33t7du39+6vWq1OPQ7Ll81mI5/PR7VanWswWavVIp1ORzqdjkajMbd+b7Ko15Bki/r3W+ZxAQARgjAA4A43BQntdjva7fbMfTWbzeh2u1EsFq/1K7B4uHK5PNf+5jlbJ5/PR6fTiU6nE9lsdm793sWMo/m4z7/fNMfjqo4LAJJLEAYA3CmVSl1bZnafWWGDpZZv3ryZS138Tr1etxk8j4bjEYDHShAGAEylUqmMPT4+Po5utzv183u9XhwfH0c2mzXzY8663e7cZ4PBfTkeAXjMBGEAwFTS6XTk8/mxr92239dNBjPIzAabr16vF4VCYdVlQEQ4HgF4/ARhAMDUrs4Ku+0KkDep1+uRTqejWCwuoLJk6vV68fLly5lm5sGiOB4BWAeCMABgavl8/tqyxmn2ARoEZleDtFnV6/UoFAqRyWRiY2Mjtra2IpfLxeHh4cz95HK52Nraio2NjchkMlEul6PZbE71/G63G4eHh5HJZMYuGjDod9BnqVSaOigcvD+D1zbo47b3rN1ux9dff33togWD5w6evyiD5W+D93B0zFKpFMfHxzfWXC6X76xrEe/vvF7DNHq93rD+QSjU7XbH/n23traiUChMfdGJQZ+FQmFY7+D4r1QqU4dPzWZzeA5tbW0Nb4OvTTLp3+8hx+O0x8XAPN6LZR1jADxCfQCACbLZbD+VSg0fNxqNfkQMb6Pfu006ne5f/bUjlUqN9TPJYMxUKtVvNBrDr7darWHf6XS6f35+PrGfTqfTz2az/YjoV6vVfqfT6Z+cnAy/Nhhjb2+vf3Bw0K9Wq/1qtdo/Pz/vn5+f9w8ODobjDW6tVqt/fn4+1sfV96fT6Uysq1ar9VOpVL9YLA7HzOfzwz7S6XS/1WqNPef8/LzfarX61Wr1Wj2jt4cYfa21Wu3GNgcHB8MaT05Orn2/1Wr19/b2xv69bzpmFvX+zuM1TKPRaNz479HpdK6dM1dv1Wp1Yt+1Wm2sbavVGvY7+r4Ui8WJ/RSLxRvfh729vWEfV8+hWf79Zjkep+13nu/FIs9hANaHIAwAmOhqENbvXw+xbgsY+v1+/+TkpB8R/YODg4l93PX82wKDVqs1dRAw+AB8tZ/z8/NrH46vhhmD0Gw0NBi89kGIdXJy0u90OsOvDdrk8/lbaxr0d1MAc1OoMun9mfffOO8KkQZj7+3t3fj8TqczDBTvCjwW9f4+9DXMavR4HBz3g2OzVqv1a7XaMJCaJgwbPH9SGDPa322B8OBYunoeDgze99HnzvLvNzDN8Xiffvv9h78XizrGAFgvgjAAYKKbgrCrAU06nb71+YOZTVc/uE4bhI1+sL0tLJimn9EPvze5+gF+0gyQq7NFbgqxrs4CuimcGMxuuS2c6PfHg5ybPoyvKggb1D4pBB01eszcFXjM6/2d92uYxvn5+bVzY5oA86bXMDr76a4a7zpObguBr9Z923E/7b/frMfjtP3O873o9+d7jAGwXuwRBgDM7ODgYOxxt9u9cX+twdeLxWKk0+kHj5tKpe78+k37A/V6veFeZrf1kc/nx2qs1WpT1fHu3btrV9OMiGsXBfj48eO1NoP9v8rl8q1jjfbdbDYfxUbklUolKpVKtFqt2Nvbm+o5s/z7z+v9neQ+r+E+Go3Gja89n89Ho9G4VtNNj1Op1J01jj632WxeOx8Hx81333134/NTqdTEC1nM4/x9SL/zfC8G/Qws6hgD4HEShAEA93I1DKtWq9faDMKkN2/e3Hucd+/eRbVajUajMVVgcdPG1qMfYCdtfD36wXcegdPoh/yr/Y1ecbPX60W73b7xdjW4m3ZD/0UplUpRr9ej1Wpdu3DCsk16fydZ5mu4LXiNiGsB8ehxOnp8TBMW7e3tjY119Xwc9NFut6NQKNx4HlSr1YUFXg8x7/diFvc9xgB4vARhAMC9XA23bpqtdHh4GNls9kFhQyqVioODgxtnqwzColndFoZ98803w/uLDgRGZwPlcrlbb1eviNlqtRZa1216vV7kcrk4Pj6Od+/ePcrA5C6P8TWMzgYcPZZPTk6G96et8+rswVFXZ0l9/fXX1674+hjej5vM+70AINkEYQDAvdy0lGr0w/bgQ/ZDZoPdpNvtRr1ej2azOXXI9uLFi7HHVwOAm4yGYoswmP2TzWaj/2Xf1qluk5ZsLtLbt2+HQc3r168nzqx7rB7ja7ga7Axqus/so9v6ivgyS2r0fO31elEulyOTycTx8fHMYy3TvN8LAJJNEAYA3NvVkOv4+Hj4gbNard6579AsBku6arVa7O3t3binz21SqdTY8qi3b9/e+MF4sH9SNpudW923GV0WuQ7evHkzDB17vV68fPlyxRXN7jG+httmOI0eF9MGQZlMZuzx58+fxx43Go1rywS73W6USqUoFAqPdunfIt4LAJJLEAYA3NtNM7Levn07XCY5r9lg5XI5crlcpNPpe+/3c3BwMHzuYIncYMlUr9eLw8PDOD4+jmKxGO/fv59L3dPodrtrEYalUqloNBrDvZfa7fbETf4fo3V4DYPa7roAxE2eP38+9vimkO3g4CDOz8+vBb3NZjMymcy9lhov2qLeCwCSSRAGADzI1WCqXq8PvzaPq/GVy+XhUsaHbHod8SUE6HQ6EfFlhkipVIqNjY3I5XLx4cOHaLVaY0HJsnz//fdLHe++0un02N5m9Xp9qmWmj8ljew2jwc5oqDwa3PR6vanC0tFZT5OCn0Eg2Gq1rs2sfAyz5K5a5HsBQPIIwgCAB8nn89c+qDabzWtXbruPdrs9DCnS6fRcAqpyuRy1Wi3Oz8/j/Pw8+v1+dDqdaDQaS70K4mgAMUvAl8vlVrqELZ/Pj10xtFwur91m5I/pNYz+W3777bc33o+Ybl+7QcgbEddmfJVKpWvts9lsnJycjO071+v1Ht2eYfN+LwBINkEYADDRNDMwRjfJn/S1q/3eZTScmBT+TNPXYDlkoVCYy0y1hxoNJrrd7lRL9Eql0rXgcRWq1epYaFgqlR7t/lK3eSyvYbAvXcT4DMpisTgW/E5zkYTR8+Wm/ftus7e3NxYMPrb9tOb9XgCQbIIwAGCiaT4UX539dZ+wZpo
      "text/plain": [
       "<Figure size 1200x900 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.hist(tracked_zmag, bins=100, label=\"tracked\", alpha=0.6, density=True)\n",
    "plt.hist(ghost_zmag, bins=100, label=\"ghosts\", alpha=0.6, density=True)\n",
    "plt.xlabel(r\"$z_{mag}$\")\n",
    "plt.ylabel(\"a.u.\")\n",
    "plt.title(\"Magnet kink position\")\n",
    "plt.legend()\n",
    "#plt.colorbar(h2[3])\n",
    "plt.show()"
   ]
  },
  {
   "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": "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": "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
}