Small updates for signUt analysis

notebooks/EWClosure.ipynb

@@ -29,7 +29,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "res, meta, _ = input_tools.read_infile(\"/scratch/dwalter/results_histmaker/230907_ew_closure/w_z_gen_dists.hdf5\")"
+    "res, meta, _ = input_tools.read_infile(\n",
+    "    \"/scratch/dwalter/results_histmaker/230907_ew_closure/w_z_gen_dists.hdf5\"\n",
+    ")"
    ]
   },
   {
@@ -79,21 +81,21 @@
     "base_process = \"WplusToMuNu\"\n",
     "# base_process = \"WminusToMuNu\"\n",
     "# base_process = \"ZToMuMu\"\n",
-    "# generator = \"winhac\" \n",
+    "# generator = \"winhac\"\n",
     "generator = \"horace\"\n",
     "generatorNLO = \"horace\"\n",
     "process = f\"{base_process}_{generator}\"\n",
     "processNLO = f\"{base_process}_{generatorNLO}\"\n",
     "var = \"ewLepEta1\"\n",
     "showFlow = False\n",
-    "flow=True\n",
+    "flow = True\n",
     "colors = mpl.colormaps[\"tab10\"]\n",
     "text_dict = {\n",
     "    \"ZToMuMu\": r\"$\\mathrm{Z}\\rightarrow\\mu\\mu$\",\n",
     "    \"WplusToMuNu\": r\"$\\mathrm{W}^+\\rightarrow\\mu\\nu$\",\n",
-    "    \"WminusToMuNu\": r\"$\\mathrm{W}^-\\rightarrow\\mu\\nu$\"\n",
+    "    \"WminusToMuNu\": r\"$\\mathrm{W}^-\\rightarrow\\mu\\nu$\",\n",
     "}\n",
-    "plotname = f\"{process}{var}\"\n"
+    "plotname = f\"{process}{var}\""
    ]
   },
   {
@@ -105,11 +107,15 @@
    "source": [
     "def load(process, order, var, appendix=\"\", idx=None):\n",
     "    proc_name = f\"{process}-{order}\"\n",
-    "    histo = res[proc_name]['output'][f\"nominal_{var}{appendix}\"].get()\n",
+    "    histo = res[proc_name][\"output\"][f\"nominal_{var}{appendix}\"].get()\n",
     "    if idx is not None:\n",
-    "        histo = histo[...,idx]\n",
-    "    histo = hh.scaleHist(histo, res[proc_name][\"dataset\"][\"xsec\"]*10e6/res[proc_name]['weight_sum'], createNew=True)\n",
-    "    return histo\n"
+    "        histo = histo[..., idx]\n",
+    "    histo = hh.scaleHist(\n",
+    "        histo,\n",
+    "        res[proc_name][\"dataset\"][\"xsec\"] * 10e6 / res[proc_name][\"weight_sum\"],\n",
+    "        createNew=True,\n",
+    "    )\n",
+    "    return histo"
    ]
   },
   {
@@ -121,7 +127,7 @@
    "source": [
     "hLO_Uncorr = load(process, \"lo-photos\", var)\n",
     "hLO_Corr = load(process, \"lo-photos\", var, appendix=\"_horacenloewCorr\", idx=1)\n",
-    "hNLO = load(processNLO, \"nlo\", var)\n"
+    "hNLO = load(processNLO, \"nlo\", var)"
    ]
   },
   {
@@ -152,61 +158,89 @@
    "outputs": [],
    "source": [
     "def plot(h1ds, ratio=False):\n",
-    "    \n",
-    "    labels = [f\"{generator} LO\",f\"{generator} \" + r\"LO $\\times$ Corr(Horace)\", f\"{generatorNLO} NLO\"]\n",
+    "\n",
+    "    labels = [\n",
+    "        f\"{generator} LO\",\n",
+    "        f\"{generator} \" + r\"LO $\\times$ Corr(Horace)\",\n",
+    "        f\"{generatorNLO} NLO\",\n",
+    "    ]\n",
     "    linestyles = [\"-\", \"--\", \":\"]\n",
-    "    \n",
+    "\n",
     "    if ratio:\n",
-    "        ylabel=f\"1./{labels[-1]}\"\n",
+    "        ylabel = f\"1./{labels[-1]}\"\n",
     "    else:\n",
-    "        ylabel=\"a.u.\"\n",
-    "    \n",
+    "        ylabel = \"a.u.\"\n",
+    "\n",
     "    ymin = None if ratio else 0\n",
     "    ymax = None\n",
     "\n",
-    "    ymax = ymax if ymax is not None else max([max(h.values(flow=showFlow)) for h in h1ds])\n",
-    "    ymin = ymin if ymin is not None else min([min(h.values(flow=showFlow)) for h in h1ds])\n",
+    "    ymax = (\n",
+    "        ymax if ymax is not None else max([max(h.values(flow=showFlow)) for h in h1ds])\n",
+    "    )\n",
+    "    ymin = (\n",
+    "        ymin if ymin is not None else min([min(h.values(flow=showFlow)) for h in h1ds])\n",
+    "    )\n",
     "    yrange = ymax - ymin\n",
-    "    ymin = ymin if ymin == 0 else ymin - yrange*0.3\n",
-    "    ymax = ymax + yrange*0.3\n",
+    "    ymin = ymin if ymin == 0 else ymin - yrange * 0.3\n",
+    "    ymax = ymax + yrange * 0.3\n",
     "\n",
-    "    fig, ax = plot_tools.figure(hNLO, \n",
-    "                      xlabel=xaxis_labels[var], ylabel=ylabel, cms_label=\"Preliminary\", \n",
-    "                      automatic_scale=False, width_scale=1.2,ylim=(ymin, ymax))\n",
+    "    fig, ax = plot_tools.figure(\n",
+    "        hNLO,\n",
+    "        xlabel=xaxis_labels[var],\n",
+    "        ylabel=ylabel,\n",
+    "        cms_label=\"Preliminary\",\n",
+    "        automatic_scale=False,\n",
+    "        width_scale=1.2,\n",
+    "        ylim=(ymin, ymax),\n",
+    "    )\n",
     "    if ratio:\n",
     "        x = h1ds[0].axes.edges[0][:-1]\n",
-    "        ax.plot([min(x), max(x)], [1,1], color=\"black\", linestyle=\"--\")\n",
-    "    \n",
+    "        ax.plot([min(x), max(x)], [1, 1], color=\"black\", linestyle=\"--\")\n",
+    "\n",
     "    for i, h1d in enumerate(h1ds):\n",
     "        x = h1d.axes.edges[0][:-1]\n",
     "        y = h1d.values(flow=flow)\n",
     "        err = np.sqrt(h1d.variances(flow=flow))\n",
     "\n",
-    "        ax.set_xlim((min(x),max(x)))\n",
-    "        ax.set_ylim((ymin,ymax))\n",
+    "        ax.set_xlim((min(x), max(x)))\n",
+    "        ax.set_ylim((ymin, ymax))\n",
     "\n",
     "        if flow:\n",
     "            # add extra bin with bin wdith of 1% of total width\n",
     "            rangex = x[-1] - x[0]\n",
     "\n",
     "            if h1d.axes[0].traits.underflow:\n",
-    "                x = np.insert(x, 0, x[0] - rangex*0.02)\n",
+    "                x = np.insert(x, 0, x[0] - rangex * 0.02)\n",
     "            if h1d.axes[0].traits.overflow:\n",
-    "                x = np.append(x, x[-1] + rangex*0.02)\n",
+    "                x = np.append(x, x[-1] + rangex * 0.02)\n",
     "\n",
     "        if showFlow:\n",
-    "            ax.set_xlim((min(x),max(x)))\n",
-    "        \n",
-    "#         ax.set_xlim(26,56)#-2.4,2.4)\n",
+    "            ax.set_xlim((min(x), max(x)))\n",
+    "\n",
+    "        #         ax.set_xlim(26,56)#-2.4,2.4)\n",
     "\n",
-    "        ax.step(x, y, color=colors(i), linestyle=linestyles[i], label=labels[i], where=\"post\")\n",
+    "        ax.step(\n",
+    "            x,\n",
+    "            y,\n",
+    "            color=colors(i),\n",
+    "            linestyle=linestyles[i],\n",
+    "            label=labels[i],\n",
+    "            where=\"post\",\n",
+    "        )\n",
     "    #     ax.fill_between(x, y - err, y + err, alpha=0.3, color=colors(i), step=\"post\")\n",
     "\n",
-    "    ax.text(1.0, 1.003, text_dict[base_process], transform=ax.transAxes, fontsize=30,\n",
-    "            verticalalignment='bottom', horizontalalignment=\"right\")\n",
+    "    ax.text(\n",
+    "        1.0,\n",
+    "        1.003,\n",
+    "        text_dict[base_process],\n",
+    "        transform=ax.transAxes,\n",
+    "        fontsize=30,\n",
+    "        verticalalignment=\"bottom\",\n",
+    "        horizontalalignment=\"right\",\n",
+    "    )\n",
     "    plot_tools.addLegend(ax, ncols=2, text_size=12)\n",
     "    if ratio:\n",
-    "        plot_name = plotname+\"_ratio\"\n",
+    "        plot_name = plotname + \"_ratio\"\n",
     "    else:\n",
     "        plot_name = plotname\n",
     "    plot_tools.save_pdf_and_png(\"./\", plot_name)"