|
|
{ "cells": [ { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "/auto/work\n" ] } ], "source": [ "%cd ..\n", "import uproot\t\n", "import numpy as np\n", "import matplotlib.pyplot as plt\n", "from matplotlib import colormaps\n", "from mpl_toolkits import mplot3d\n", "import awkward as ak\n", "from scipy.optimize import curve_fit\n", "from scipy import stats\n", "from methods.fit_linear_regression_model import fit_linear_regression_model\n", "import sklearn\n", "import seaborn as sns\n", "import pandas as pd\n", "%matplotlib inline" ] }, { "cell_type": "code", "execution_count": 11, "metadata": {}, "outputs": [], "source": [ "def round(n, k):\n", " # function to round number 'n' up/down to nearest 'k'\n", " # use positive k to round up\n", " # use negative k to round down\n", "\n", " return n - n % k" ] }, { "cell_type": "code", "execution_count": 12, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "41978 8523\n", "49865\n" ] } ], "source": [ "file = uproot.open(\n", " \"/work/cetin/Projektpraktikum/tracking_losses_ntuple_B_rad_length_beginVelo2endVelo.root:PrDebugTrackingLosses.PrDebugTrackingTool/Tuple;1\"\n", ")\n", "\n", "# selektiere nur elektronen von B->K*ee\n", "allcolumns = file.arrays()\n", "found = allcolumns[(allcolumns.isElectron) & (~allcolumns.lost) &\n", " (allcolumns.fromB)]\n", "lost = allcolumns[(allcolumns.isElectron) & (allcolumns.lost) &\n", " (allcolumns.fromB)]\n", "\n", "electrons = allcolumns[(allcolumns.isElectron)\n", " & (allcolumns.fromB)\n", " & (allcolumns.eta <= 5.0)\n", " & (allcolumns.eta >= 1.5)\n", " & (np.abs(allcolumns.phi) < 3.142)]\n", "\n", "print(ak.num(found, axis=0), ak.num(lost, axis=0))\n", "print(ak.num(electrons, axis=0))\n", "# ak.count(found, axis=None)" ] }, { "cell_type": "code", "execution_count": 13, "metadata": {}, "outputs": [], "source": [ "# variables\n", "\n", "eta_a = ak.to_numpy(electrons[\"eta\"])\n", "phi_a = ak.to_numpy(electrons[\"phi\"])\n", "rad_length_frac_a = ak.to_numpy(electrons[\"rad_length_frac\"])" ] }, { "cell_type": "code", "execution_count": 14, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAjYAAAHLCAYAAADbUtJvAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAAAseklEQVR4nO3dwW/j6H3/8Y/bYB2nxqzsySW3DAX00suAmr3Gh5GQQ462Zm/ZFsFICHz4AYNWgnsZz6WCjF4NhJpDF7nNiHPsoRW36OTYGbHTP0Cc9pQCzdrMwIjjxSb6HTbkWrYky6RoUY/fL0DYHVHk85CPRH79PA+/XBkOh0MBAAAY4M8WXQEAAIB5IbABAADGILABAADGILABAADGILABAADGILABAADGILABAADGILABAADGILDBRL7vq9PpLLoat14Yhup0OgqCYNFVMUoQBOp0OgrDcNFVwRxx3gKBDS4JgkDValWlUkmO44y8v7GxMdeTRhbbnJXv+2o2myqVSiqVSjde/iw6nY7u3buner2e+gIc7W+1WlWxWNTBwcF8KpnQItv+4OBAxWJR9XpdR0dHN1Zumn32PE/FYlErKytaWVlRqVSS67qXPue6rkqlUvwZ3/enbndev4PovLGxsaGNjQ1Vq9UbDRonnbfyftyQgSEwgaShbdvxv/v9/lDSsNFoJNreYDAYHh8fj7yXdptpReWf389FGneMGo3GUNKw3+8n3m6/3x8WCoWRbdZqtcTbu648tv3Ozs5Q0nAwGNxYmWn3+fj4eChpKGnoOM7EzzmOM9Les9Yr6e8g+n7t7OwMa7XasFAoDCUNC4XCpXbP2rj9yOtxQzYIbDDRvH+wtm3f6EVkVnk6MY07Ru12O3Vgs7OzM7QsK231Estj29dqtRsPbOYhCnTL5fLEz9RqtWG73b7WdtP8Dsrl8qXjWC6XFxK4TtqPPB43ZIOhKNyIarV6ZdfubZflMVrksaft52tvb0/SN0Msk47ry5cvVavVbqQ+vu+rWq3KsqyR99vtdrw8D/J23JAdAhsDuK6rSqUiz/PU6XS0sbGher0eLz84OFC9Xo/Hg8eN74dhqHq9Hr/Gzb9wXVfValXVavXSsmlluK4bn0jq9frIhW7aNqM6NZtNVSqVeB/PL+90OvGYued58Rj5uO1dV3TCrlQqKhaLajabqcqOjm2xWLx0sZ92jCJHR0dxmbPuY6fTUbVaVRAE8RyEarUaH8d5f3eidjrfvnlv+6isaG7I+f2PTPsunDetjSft83X3pVAoaGdnR5LUarUuLXddV+VyWYVC4dr1H+eqtrBte2wwYNt2XN/z5vm7muW8FcnbcUOGFt1lhHS63e7QsqyhpGGtVhs2Go2hbdtx12jU/Rrp9XpDScNerxe/NxgMhoVCYeS9aPgj2s5gMIjfu9iVO0sZ0WfOd1dP22Y0Zn9++MVxnKGkuKt4MBjE8yTK5fKw0WgM+/1+PMQwa5eyxnQl9/v9kTp1u934GCcp27btuEt+MBjE4/2WZcXbHHeMhsPhyDFqt9uJ9tGyrEtDUfP67liWNVLnQqFwaT5PHts++my5XB7WarWh4zhD27Yv1emq70LkqjaetM9J9uX89i/OYbFte6SNZq3/pN/BVW0xSVTH83Na5vm7muW8NalOeT5uSI/AxgDRj3ncWHa5XB65yEST6M5/dmdnZ+y487gf7LgL0SxlTLpoT9qmbdtj6xRdeKLtRCebiyeLcducZNx+2rZ9aU5LNCEyOiHOWnb0ufPbGze/46rA5uKkx+vs47jA5vy2k353bNu+tP/RNqPjlNe2j9qg2+2OvB8Fe9FF7jrfhavaeFL9kuzLuDks0cX+vFnqH5U17ncwS1uM0263L33n5vm7us5567y8Hzekx1CUAaKu008++eTSsm63q36/H//77du3khTfhhkEQTwckdRVZVxXEATyfT/uyj4vGiY4fzundLm7W1Li23ij8lutVjxscL4bPNq/Wct+8+bNpeXR9q4z/+DBgweXyk2b2ybtd8f3fZXL5ZH1Go2GhsPh2ONylUW0/cWyonJ6vd7M34V5tfF19iUaFjk/PNhut+O5JNL1v8vnJWmL81qtlrrdbuK6TDsWac5beT9uSO87i64A5mfciaBQKKhQKMh1Xb148eLSBSy6MF6c+HfdcqeVcV3TLgTRxT3LZHVR+edPymlEx8PzvEsnu3Env0VI8t2JjlOSAGaSRbe9pDhQiy5Q0tXfhUW0cblclmVZcaLBWq2mly9f6v379/Fn0nyX07RFvV7X8+fPR/Z9nr+rNOetPB83zAc9NoYLgkClUklBEKjb7arRaFxaLiXv3ZiljKTG9fhEF9HNzc25lDFOdEzmdfLZ2dlRuVxWq9WS53kKw1DtdluNRiNVQJm1Wb87WZykF9X258uJLn7S1fu4qDaOeh/a7bZc19WDBw9GAs15tNF126LT6ahYLMYTdedZl4vbSnreyuNxw/wQ2BiuUqloc3NzYrARnXTPDznMu4zriv7KG3cHQXSyKBaLcylrnOiYjMtOOqleV+l2uyqXy3G693a7Hd8Om1dXtWvUTpP+qk1yUVh0258v55NPPrnWd2ERbVyr1eIhycePH1+6oyvNdzlJW7iuqzAMx35n5vm7Snveyttxw3wR2BgsusX3/F8i0Q8r+ksn6hqd9Mycq+bJzFLGdbYnfXNSsW073vZ5b9++VaFQyDTXRDQU0Ww2L3UrJ03/X61W416PRqMxdXji4jH68ssvJaXrVTs6OrrW+tf97lw8iTebzUt/lS5D20vfXJAsy4p7YaTZvgvXaeN5Oj835GIvSZrv8nXbwvM8vXnzZmxQ0+l05vq7SnvekvJz3DB/BDYGiH7EF3/M0YXFdV11Oh11Op24C9b3/fivkehEVCqV5HmegiCIPxcEQZwbYlzAMksZYRjGf6E4jhNP/Ju0Tembv34LhcLIX1JR9/7z58/jC24Wz/kpFAojx6Rarerg4ECVSkWDwSA+6c1adnThj7bT6XRG8rtEJh2jScIwTL3/ab87UY9EpVJRtVqN890Ui8W4jfLa9tF2zl98giCQ4zhxL9Ss34VZ23jSPqfZl+giOe5iOWv9J5m1LaJ8L2EYqtlsxq96va5SqSTLsub6u7q4rWnnrUnycNyQkUXfloV0zucisSzr0i3B0bNPLMuKb2+MnuVy/nZHx3Hi7UTp76N1BoPBsN/vx/kl9KdbMaNbHmctw7btYaFQiPNATNvmcPjN7cXRLZ21Wm1Yq9VGbr/s9/vx7ZOWZQ17vd7w+Pg4vs1WY24ZPe98foxx5Ue3q447ttcpOzqW0fvnX5ZljZR58Rh1u934FtPo9tOL5UxLWX9xH2u1Wnwb87y+O91uNz4WF3OBTNqvRbd9VEaj0RiWy+WRcsY922jad2E4nK2NJ+3zPPYlypMzyVXf5Wm/g6va4nxumHGvi7dRz+t3NRxefd66yiKPG7KzMhwOh3OJkACM5fu+Xrx4ob29PR0dHY30tHS7XRWLxbnNT8Ji0MZAfnC7N5Ch6M6i4+Pj+Pbp8yzLIs36kqONgXxhjg2QoWj+xuPHj0fmW0Q5NBzHYSLhkqONgXxhKArI2MHBgVqt1sgEXdu21W63r5yEiOVAGwP5QWAD3JBo3kWek/IhHdoYWDwCGwAAYAzm2AAAAGMs9V1Rv/nNb/Qv//Iv+uEPf6i1tbVFVwcAAMzg9PRU//3f/60f//jH+v73vz/XbS91YPPP//zP+uu//utFVwMAACTw+eef67PPPpvrNpc6sPnBD34gSXr+/Hmmz2XZ3t7Wq1evMtu+KWWcnJxoa2tLr1+/1vr6eiZlmHCcbqKMm2gLKfv9oC1mZ8Kxoi1uTxm+7+vx48fxdXyeMgtsfv7zn+vBgwf62c9+llUR+u53vytJ+su//MtMA5u1tbXMH2hnQhkfPnyQJN2/f1937tzJpAwTjtNNlHETbSFlvx+0xexMOFa0xe0p4+TkRNK31/F5ymTy8Pv37+U4DinEAQDAjcqkx+bevXtyHIdcDgAA4EZlNhT1+PHjrDYNAAAwVmZ5bH784x/r3bt3WW0eAAD "text/plain": [ "<Figure size 640x480 with 1 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "plt.hist(\n", " rad_length_frac_a,\n", " bins=100,\n", " density=True,\n", " alpha=0.5,\n", " color=\"blue\",\n", " histtype=\"bar\",\n", " range=[0, 1],\n", ")\n", "plt.xlim(0, 1)\n", "# plt.yscale(\"log\")\n", "plt.title(\"radiation length fraction beginVelo2endVelo\")\n", "plt.xlabel(f\"$x/X_0$\")\n", "plt.ylabel(\"a.u.\")\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": 15, "metadata": {}, "outputs": [], "source": [ "df = pd.DataFrame({\n", " \"phi\": phi_a * 90.0 / np.pi,\n", " \"eta\": eta_a * 2.0,\n", " \"rad_length_frac\": rad_length_frac_a,\n", "})\n", "df = df.round({\"phi\": 0, \"eta\": 1, \"rad_length_frac\": 4})" ] }, { "cell_type": "code", "execution_count": 16, "metadata": {}, "outputs": [], "source": [ "df_pivoted = df.pivot_table(\n", " index=\"eta\",\n", " columns=\"phi\",\n", " values=\"rad_length_frac\",\n", " margins=False,\n", " # fill_value=0,\n", ")" ] }, { "cell_type": "code", "execution_count": 17, "metadata": {}, "outputs": [ { "data": { "text/html": [ "<div>\n", "<style scoped>\n", " .dataframe tbody tr th:only-of-type {\n", " vertical-align: middle;\n", " }\n", "\n", " .dataframe tbody tr th {\n", " vertical-align: top;\n", " }\n", "\n", " .dataframe thead th {\n", " text-align: right;\n", " }\n", "</style>\n", "<table border=\"1\" class=\"dataframe\">\n", " <thead>\n", " <tr style=\"text-align: right;\">\n", " <th>phi</th>\n", " <th>-90.0</th>\n", " <th>-89.0</th>\n", " <th>-88.0</th>\n", " <th>-87.0</th>\n", " <th>-86.0</th>\n", " <th>-85.0</th>\n", " <th>-84.0</th>\n", " <th>-83.0</th>\n", " <th>-82.0</th>\n", " <th>-81.0</th>\n", " <th>...</th>\n", " <th>81.0</th>\n", " <th>82.0</th>\n", " <th>83.0</th>\n", " <th>84.0</th>\n", " <th>85.0</th>\n", " <th>86.0</th>\n", " <th>87.0</th>\n", " <th>88.0</th>\n", " <th>89.0</th>\n", " <th>90.0</th>\n", " </tr>\n", " <tr>\n", " <th>eta</th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " <th></th>\n", " </tr>\n", " </thead>\n", " <tbody>\n", " <tr>\n", " <th>3.1</th>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>3.2</th>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>3.3</th>\n", " <td>NaN</td>\n", " <td>0.30455</td>\n", " <td>NaN</td>\n", " <td>0.526900</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.14755</td>\n", " <td>NaN</td>\n", " <td>0.135100</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>0.1247</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>3.4</th>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.459800</td>\n", " <td>0.1691</td>\n", " <td>0.1710</td>\n", " <td>0.26650</td>\n", " <td>NaN</td>\n", " <td>0.291200</td>\n", " <td>0.1778</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.14020</td>\n", " <td>0.1775</td>\n", " <td>0.5234</td>\n", " <td>0.1498</td>\n", " <td>NaN</td>\n", " <td>0.14810</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>3.5</th>\n", " <td>NaN</td>\n", " <td>0.45340</td>\n", " <td>0.35060</td>\n", " <td>0.447200</td>\n", " <td>NaN</td>\n", " <td>0.1885</td>\n", " <td>NaN</td>\n", " <td>0.1711</td>\n", " <td>0.243100</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>0.2161</td>\n", " <td>0.1313</td>\n", " <td>NaN</td>\n", " <td>0.16210</td>\n", " <td>NaN</td>\n", " <td>0.4284</td>\n", " <td>0.4008</td>\n", " <td>NaN</td>\n", " <td>0.24475</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>...</th>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " <td>...</td>\n", " </tr>\n", " <tr>\n", " <th>9.6</th>\n", " <td>NaN</td>\n", " <td>0.06550</td>\n", " <td>0.09655</td>\n", " <td>0.181325</td>\n", " <td>0.1296</td>\n", " <td>NaN</td>\n", " <td>0.25035</td>\n", " <td>0.0707</td>\n", " <td>NaN</td>\n", " <td>0.2469</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>0.0692</td>\n", " <td>0.2822</td>\n", " <td>0.26520</td>\n", " <td>0.1244</td>\n", " <td>NaN</td>\n", " <td>0.1084</td>\n", " <td>0.1665</td>\n", " <td>0.06750</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>9.7</th>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.070000</td>\n", " <td>0.0816</td>\n", " <td>0.0878</td>\n", " <td>0.07350</td>\n", " <td>0.0864</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>0.1993</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.14480</td>\n", " <td>0.0652</td>\n", " <td>0.0755</td>\n", " <td>NaN</td>\n", " <td>0.0796</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>9.8</th>\n", " <td>0.0781</td>\n", " <td>0.09960</td>\n", " <td>0.08180</td>\n", " <td>0.083267</td>\n", " <td>0.0779</td>\n", " <td>NaN</td>\n", " <td>0.08810</td>\n", " <td>0.1684</td>\n", " <td>0.140867</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>0.0977</td>\n", " <td>NaN</td>\n", " <td>0.13015</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.0821</td>\n", " <td>0.0646</td>\n", " <td>0.08440</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>9.9</th>\n", " <td>NaN</td>\n", " <td>0.09210</td>\n", " <td>0.08210</td>\n", " <td>0.071900</td>\n", " <td>NaN</td>\n", " <td>0.0936</td>\n", " <td>0.32600</td>\n", " <td>0.0748</td>\n", " <td>0.193300</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>0.0727</td>\n", " <td>0.1494</td>\n", " <td>0.0353</td>\n", " <td>NaN</td>\n", " <td>0.3264</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.0844</td>\n", " <td>0.07660</td>\n", " <td>NaN</td>\n", " </tr>\n", " <tr>\n", " <th>10.0</th>\n", " <td>NaN</td>\n", " <td>0.05840</td>\n", " <td>NaN</td>\n", " <td>0.070600</td>\n", " <td>NaN</td>\n", " <td>0.0578</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>...</td>\n", " <td>NaN</td>\n", " <td>0.0785</td>\n", " <td>0.0915</td>\n", " <td>0.03070</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>NaN</td>\n", " <td>0.07090</td>\n", " <td>NaN</td>\n", " </tr>\n", " </tbody>\n", "</table>\n", "<p>70 rows × 181 columns</p>\n", "</div>" ], "text/plain": [ "phi -90.0 -89.0 -88.0 -87.0 -86.0 -85.0 -84.0 -83.0 \\\n", "eta \n", "3.1 NaN NaN NaN NaN NaN NaN NaN NaN \n", "3.2 NaN NaN NaN NaN NaN NaN NaN NaN \n", "3.3 NaN 0.30455 NaN 0.526900 NaN NaN 0.14755 NaN \n", "3.4 NaN NaN NaN 0.459800 0.1691 0.1710 0.26650 NaN \n", "3.5 NaN 0.45340 0.35060 0.447200 NaN 0.1885 NaN 0.1711 \n", "... ... ... ... ... ... ... ... ... \n", "9.6 NaN 0.06550 0.09655 0.181325 0.1296 NaN 0.25035 0.0707 \n", "9.7 NaN NaN NaN 0.070000 0.0816 0.0878 0.07350 0.0864 \n", "9.8 0.0781 0.09960 0.08180 0.083267 0.0779 NaN 0.08810 0.1684 \n", "9.9 NaN 0.09210 0.08210 0.071900 NaN 0.0936 0.32600 0.0748 \n", "10.0 NaN 0.05840 NaN 0.070600 NaN 0.0578 NaN NaN \n", "\n", "phi -82.0 -81.0 ... 81.0 82.0 83.0 84.0 85.0 86.0 \\\n", "eta ... \n", "3.1 NaN NaN ... NaN NaN NaN NaN NaN NaN \n", "3.2 NaN NaN ... NaN NaN NaN NaN NaN NaN \n", "3.3 0.135100 NaN ... 0.1247 NaN NaN NaN NaN NaN \n", "3.4 0.291200 0.1778 ... NaN NaN NaN 0.14020 0.1775 0.5234 \n", "3.5 0.243100 NaN ... 0.2161 0.1313 NaN 0.16210 NaN 0.4284 \n", "... ... ... ... ... ... ... ... ... ... \n", "9.6 NaN 0.2469 ... NaN 0.0692 0.2822 0.26520 0.1244 NaN \n", "9.7 NaN NaN ... 0.1993 NaN NaN 0.14480 0.0652 0.0755 \n", "9.8 0.140867 NaN ... NaN 0.0977 NaN 0.13015 NaN NaN \n", "9.9 0.193300 NaN ... 0.0727 0.1494 0.0353 NaN 0.3264 NaN \n", "10.0 NaN NaN ... NaN 0.0785 0.0915 0.03070 NaN NaN \n", "\n", "phi 87.0 88.0 89.0 90.0 \n", "eta \n", "3.1 NaN NaN NaN NaN \n", "3.2 NaN NaN NaN NaN \n", "3.3 NaN NaN NaN NaN \n", "3.4 0.1498 NaN 0.14810 NaN \n", "3.5 0.4008 NaN 0.24475 NaN \n", "... ... ... ... ... \n", "9.6 0.1084 0.1665 0.06750 NaN \n", "9.7 NaN 0.0796 NaN NaN \n", "9.8 0.0821 0.0646 0.08440 NaN \n", "9.9 NaN 0.0844 0.07660 NaN \n", "10.0 NaN NaN 0.07090 NaN \n", "\n", "[70 rows x 181 columns]" ] }, "execution_count": 17, "metadata": {}, "output_type": "execute_result" } ], "source": [ "df_pivoted" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] }, { "cell_type": "code", "execution_count": 18, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAABdkAAAJRCAYAAAC5lb6oAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOzdeXhV5b0+/DvzzrwzEDIBYYdJZDIBResMcbZORK2ttXUgKq09rT3k2HqK9NRSaHvao1IbrHawg0qcWufEGRGFROY5mzCFkHFnzs6w9/tH3/ADyf0NbGa9P9fFdU69WWs961nPtFYWK0F+v98PERERERERERERERE5bMEnugAiIiIiIiIiIiIiIqcqPWQXEREREREREREREQmQHrKLiIiIiIiIiIiIiARID9lFRERERERERERERAKkh+wiIiIiIiIiIiIiIgHSQ3YRERERERERERERkQDpIbuIiIiIiIiIiIiISID0kF1EREREREREREREJEB6yC4iIiIiIiIiIiIiEiA9ZBcREREROYW53e4TXQQRERERkROuvLwc+fn5KCwsREFBAYqLiwfcJi8vD0FBQf3+KS0tPeRjhx5JwUVEREREPs/tdqO0tBQ33ngjnE7niS7OF15+fj7efvtt1bWIiIiIfGm53W7k5uairKwMOTk5AIDs7Gw0NDRg5syZdBu324358+cfsJauqKjAggULMH369EM+vt5kFxERETkFFBcXIzc3d99bFdnZ2ViwYIG5TWlpKfLz8/dtk5CQgMLCQng8HgCAx+NBYWEhEhIS9v2dgoIClJeXH7Sv8vJyFBQUIDc3FwkJCcjNzd33lsiiRYuQn58PAFiwYAGys7NRUFCAhoaGwzrH0tJSZGdn7ytLbm5uv2+f7F8Xubm5KC8vP6h++srY92f//fa9kRJInfZX5ry8PGRnZ+87Tn5+/mG99XIkysvL4XK59t0UuN1uFBYWHlAP+5/T/nWckJBw3MopIiIiInIsFRQUYPr06fsesAPY90Y7U1pairKyMsyePRszZ87c9ycpKemwHrADAPwiIiIicsoA4AfgLysrO+RtXC6XH4C/qKio33z+/Pl+AP6cnJyDssbGRv+MGTP8APwzZ870V1RU7MsqKir806dP9wPwT58+fd9/7/v7+//dQ9XY2LjvHFl5/X6/v6ioyO90Og/6733bsmPn5OT4S0pK+t3mcOrU7/f7Z86c6QfgX7x48UFl+3ydHCszZ8486Hz8/v93DWbMmNFv1t+1FhERERE5FfXdQ8yfP/+A/15WVjbgfUV/cnJyDnsbvckuIiIicgo6nE+D9P3dxMTEw86nTZuG4uJiFBUVoaioCC6Xa1/mcrlQUlKCGTNmHPDWOjvOoZZ19uzZAIDFixfTv1dWVoYHHnjgsPdvbXM4dZqXl4dFixahrKwMM2bMOCCbOXMmSkpK9r3lfiyVlpb2+5ZN33l+/k314uJilJeXo6ys7JiWS0RERETk87xeL5qbm4/oj9frPWi/K1asAIAD7lUA7HurvaSk5JDL6PF4UF5ejhtvvPGwzk0P2UVERESkX4WFhSgvL0dOTg79jiEAPPHEE0f1uPs/IO7v0zUA8Nxzz5llYmbMmHH4//Tzc4qLi/c93N7/n6Pury8rLS09pF+4FGg52Lnk5OQgJycHHo9n34P28vJyFBYWHtZNhoiIiIjIgIKCDunPPIcD8fHxR/Rn3rx5Bx3e7XYD4C/N9OWH4rnnnkNOTs5h/74jPWQXERERkX71fct7oDfGnU5nv3/H4/GgoKAACQkJSEhIML+H+Pn99b0d3t8iuu/h8uEufIuLi/d9j/5IFBYWAsC+79AzN9100wF//2h79tlnzTrty4qKiuDxeJCfn4/Fixcf9IaPiIiIiMjx8ACApiP80999R0VFBQD+L2oP5x5g8eLF+9bxh0MP2UVERETkIPu/Qc7e1t7f5z+ZAvy/h8vz58+Hy+XCokWLDvnzKfPnzwfQ/4PxefPmHfID+89vdzT0vQkz0MPqvtx6c6bv7fLP/xAiPz8fCQkJdFuPxwO3221em75/4lpcXIxp06Zh/vz5h3QtRURERESOhQgAcUf4JyIi4qD9ZmdnA8ABn7Dc36G+ZNL3r0D7u7cZiB6yi4iIiHxJ3HXXXfveKt//T38PrPd/uBvom88FBQUoKirCzJkzUVZWBpfLhdLS0oO+E94fl8u171Mo+z8cd7vdcLvdA37yJS8vD9nZ2fv+BAUF0U/PHI7993GoD9kB/qA9JycH8+fPx+TJk7Fo0SIA/663goKCfT+c6M9zzz034Bs2+/+LAJfLFdDNgoiIiIjIgEKCj9+ffvStmdkb64d6P1NaWgqXyxXQ/Y8esouIiIh8STzxxBNobGw86E9RUZG5XaCfWPn8W9N9D/MP9ZvgfW/C9z18Bv79hvuh/MLTkpISVFRUHPDneH8mZf/P2QxUh32fnikoKEBhYSGmT59ufnO+74cXA+k7Z+sHDAUFBViwYAEKCwuP2ffjRURERESOlcmTJwM4+MWWvv+dm5t7SPt59tlnA34xRQ/ZRUREROQgh/IW9uHqe/v8UPc3ffp0uFwueDyefQ/aA/2Fpy6XK6BPzHze/j84GOg8DudfA/TVjdPpHPDvut1uJCYmDvhN+r5f0OpyueB2u/t90J6fnw+n04nZs2dj/vz5mDdv3iH9SwMRERERkX1Cgo7fn344nU7k5OQc9DJP37q27zOKAykuLg7oe+yAHrKLiIiISD/2f5i8YsWKo7LPvofCh/NG+f7fdS8uLsbkyZMP+xee9pk9e3bA2+7vUL61Dvy/t8ddLteAxz3UfQL/fot9oB8Y9H3r/e233z7gF6Duz+12o7i4+IB93XTTTfu+hy8iIiIicqp44oknUFpaesB6ev78+Zg/f/6+tbjb7UZ2dna/L5UUFxfve1gfCD1kFxEREfmCKS8vPyrfH+97Y3zx4sVHvC/g/30yZcqUKYdVBqfTCbfbjbvuuuuovI1eXl5+2G/n71+nfQ+hB6qXZ5999oC/byksLEROTs4hXbfi4mLzn7F6PB7k5+dj8eLFcDqd+67j/p/dAQ78IUCfnJwclJaWBvyJIBERERH5EjrB32QH/r2OLSsrQ2FhIQoLC5Gfn4/CwkLMnj1739/xeDxoaGjod6377LPPHvIb7/3RQ3YRERGRL5jCwsKj8v3xvrc+SktLD3pAuz+Px3NID7/7Pl1yuN853P8b7Efjl3feddddSExMPKxt9q/TGTNm7HsYzR6KFxcXo7y8HNOnTx+wzMXFxcjLy0NBQcG+X+wK9P9We2lp6YC/9HXatGkoKira9xbO/r8Adf9vri9fvvygN+z76qWhocE8hoiIiIjIySYnJweLFy/G/PnzsXjx4oM+M5mTk4PGxsZ+1+eLFy8e8HdVWfSQXUREROQU1N8DWLfbjby8PDQ0NBzw8LTv77IHp31vcnx+n06nE2VlZZg+fToKCgqQn59/wN/p+1Z631si+2/3+f253W4UFRUF9FZ83+L4cL7F3t/bKX314/F4+v18y+HUaVlZGWbOnIlp06Yd9MtCi4uLkZ+fj5kzZ9LzdbvdWLBgAYqLi9HQ0IDp06fve3heVFSEBQsW9PuDgIE+FZOXlweXy3XQg/i8vDwAwLx58/b9N4/HQ3/YcLS+wy8iIiIiXwKhQcfvz0kq9EQXQEREREQGVlxcfMCbFXl5efveVPZ4PAc8FO37J5GlpaUoKira98C5sLAQHo9n3ydYPB4P5s2bt+8tdbfbjYKCAhQUFOzbt8vlQklJyb7j5+bmwuPxwOVyweVyIT8//6BfMNT35nnfNxD73gAvKysL6JvofZ88sR4uf75+cnNzD/ie4v511PewPpA63V9RURHy8/Mxf/58FBYW7qvTvl+6ZL1xXl5ejnnz5mHmzJn7PifT95b/okWL8Pbbb/dbV+Xl5f1+J7K0tHTfm/Aul+uAT8rsf57l5eXIz89HQUEBsrOz8dxzz/VbvqPxLyFERERERL4sgvx+v/9EF0JERERERGyLFi2Cx+Pp94F/IPreuN//dqC0tBR5eXnQLYKIiIi "text/plain": [ "<Figure size 2100x700 with 2 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "fig = plt.figure(figsize=(21, 7))\n", "ax = sns.heatmap(\n", " df_pivoted,\n", " robust=True,\n", " square=True,\n", " cmap=colormaps[\"rainbow\"],\n", " xticklabels=False,\n", " yticklabels=False,\n", " vmax=0.7,\n", ")\n", "# ax.set_yticks([5, 15, 25, 35], [2, 3, 4, 5])\n", "ax.set_yticks([10, 30, 50, 70], [2, 3, 4, 5])\n", "ax.set_xticks([39, 89, 139], [-100, 0, 100]) # ([79, 179, 279], [-100, 0, 100])\n", "ax.set_xlabel(f\"$\\phi$ [deg]\")\n", "ax.set_ylabel(f\"$\\eta$\")\n", "\n", "# ax.set_yticklabels([])\n", "ax.invert_yaxis()\n", "ax.set_title(\"LHCb VELO $x/X_0$\")\n", "plt.show()" ] } ], "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}
|
