diff --git a/Antares/.ipynb_checkpoints/Analysis AAfit-checkpoint.ipynb b/Antares/.ipynb_checkpoints/Analysis AAfit-checkpoint.ipynb new file mode 100644 index 0000000000000000000000000000000000000000..57fc44c5a3786f322ab140c9ef3657c730bdb09a --- /dev/null +++ b/Antares/.ipynb_checkpoints/Analysis AAfit-checkpoint.ipynb @@ -0,0 +1,11723 @@ +{ + "cells": [ + { + "cell_type": "code", + "execution_count": 1, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "%matplotlib notebook\n", + "%load_ext autoreload\n", + "%autoreload 2\n", + "\n", + "import warnings\n", + "warnings.filterwarnings('ignore')\n", + "import resource\n", + "soft, hard = resource.getrlimit(resource.RLIMIT_NOFILE)\n", + "resource.setrlimit(resource.RLIMIT_NOFILE, (hard, hard))" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Welcome to JupyROOT 6.14/00\n" + ] + } + ], + "source": [ + "#import __future__\n", + "import numpy as np\n", + "import sys\n", + "from root_numpy import root2array, tree2array\n", + "from root_numpy import testdata\n", + "import ROOT\n", + "import matplotlib.pylab as plt\n", + "import matplotlib as mpl\n", + "import matplotlib.colors as colors\n", + "from scipy.interpolate import interp1d\n", + "import h5py\n", + "from mpl_toolkits.axes_grid1 import make_axes_locatable\n", + "from matplotlib import rc\n", + "rc('font',**{'family':'sans-serif','sans-serif':['Helvetica']})\n", + "## for Palatino and other serif fonts use:\n", + "#rc('font',**{'family':'serif','serif':['Palatino']})\n", + "rc('text', usetex=True)" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "anumu = ROOT.TFile('../../Ntuples/anumu_dm_2000GeV.root')\n", + "numu = ROOT.TFile('../../Ntuples/numu_dm_2000GeV.root')\n", + "atmu = ROOT.TFile('../../Ntuples/mupage_sun.root')\n", + "data = ROOT.TFile('../../Ntuples/nTupla_DATA_CUTTED.root')\n", + "\n", + "\n" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "# load Ntuples\n", + "anu = anumu.Get('antree')\n", + "nu = numu.Get('antree')\n", + "mu = atmu.Get('antree')\n", + "da = data.Get('antree')\n", + "# and convert the TTree into an array\n", + "anu_arr = tree2array(anu)\n", + "nu_arr= tree2array(nu)\n", + "mu_arr= tree2array(mu)\n", + "da_arr= tree2array(da)" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "h5f = h5py.File('../../datafiles/dm2bb_nu_1000.0_GeV.h5','r')\n", + "nu_w_bb_2000 = h5f['w_bb_1000.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2bb_anu_1000.0_GeV.h5','r')\n", + "anu_w_bb_2000 = h5f['w_bb_1000.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2tt_nu_1000.0_GeV.h5','r')\n", + "nu_w_tt_2000 = h5f['w_tt_1000.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2tt_anu_1000.0_GeV.h5','r')\n", + "anu_w_tt_2000 = h5f['w_tt_1000.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2WW_nu_1000.0_GeV.h5','r')\n", + "nu_w_WW_2000 = h5f['w_WW_1000.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2WW_anu_1000.0_GeV.h5','r')\n", + "anu_w_WW_2000 = h5f['w_WW_1000.0'][:]\n", + "h5f.close()\n" + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "h5f = h5py.File('../../datafiles/dm2bb_nu_100.0_GeV.h5','r')\n", + "nu_w_bb_100 = h5f['w_bb_100.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2bb_anu_100.0_GeV.h5','r')\n", + "anu_w_bb_100 = h5f['w_bb_100.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2tt_nu_100.0_GeV.h5','r')\n", + "nu_w_tt_100 = h5f['w_tt_100.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2tt_anu_100.0_GeV.h5','r')\n", + "anu_w_tt_100 = h5f['w_tt_100.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2WW_nu_100.0_GeV.h5','r')\n", + "nu_w_WW_100 = h5f['w_WW_100.0'][:]\n", + "h5f.close()\n", + "h5f = h5py.File('../../datafiles/dm2WW_anu_100.0_GeV.h5','r')\n", + "anu_w_WW_100 = h5f['w_WW_100.0'][:]\n", + "h5f.close()" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [] + }, + { + "cell_type": "code", + "execution_count": 7, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [], + "source": [ + "#dates\n", + "ymd_min, ymd_max2012, ymd_max2016 = 20072020, 20122020, 20162020 \n", + "#direction \n", + "cos_min, cos_max = 0., 1.\n", + "#quality\n", + "beta_max = 1.\n", + "# runs\n", + "run_min, run_max = 25800,68170\n", + "# lambda \n", + "lam_min, lam_max = -5.3, 0.\n", + "\n", + "# define weights\n", + "w_bg_anu = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*(anu_arr[\"mc_wH\"]+anu_arr[\"mc_wS\"])\n", + "w_bg_nu = nu_arr[\"mc_rbr\"]* nu_arr[\"mc_wc\"] *(nu_arr[\"mc_wH\"]+ nu_arr[\"mc_wS\"])\n", + "# define cuts\n", + "mu_cuts2012 = (np.cos(mu_arr[\"a_th\"])>cos_min)&(mu_arr[\"run\"]>run_min)&\\\n", + " (mu_arr[\"ymd\"]>ymd_min)&(mu_arr[\"ymd\"]lam_min)&\\\n", + "\n", + "mu_cuts2016 = (np.cos(mu_arr[\"a_th\"])>cos_min)&(mu_arr[\"run\"]>run_min)&\\\n", + " (mu_arr[\"ymd\"]>ymd_min)&(mu_arr[\"ymd\"]lam_min)&\\\n", + "\n", + " \n", + "nu_cuts2012 = (nu_arr[\"ymd\"]>ymd_min)&(nu_arr[\"ymd\"]cos_min)&(np.cos(nu_arr[\"a_th\"])lam_min)\n", + " \n", + "nu_cuts2016 = (nu_arr[\"ymd\"]>ymd_min)&(nu_arr[\"ymd\"]cos_min)&(np.cos(nu_arr[\"a_th\"])lam_min)\n", + " \n", + " \n", + "anu_cuts2012 = (anu_arr[\"ymd\"]>ymd_min)&(anu_arr[\"ymd\"]cos_min)&(np.cos(anu_arr[\"a_th\"])lam_min)\n", + "\n", + "anu_cuts2016 = (anu_arr[\"ymd\"]>ymd_min)&(anu_arr[\"ymd\"]cos_min)&(np.cos(anu_arr[\"a_th\"])lam_min)\n", + " \n", + "da_cuts2012 = (da_arr[\"ymd\"]>ymd_min)&(da_arr[\"ymd\"]cos_min)&(np.cos(da_arr[\"a_th\"])lam_min)\n", + "\n", + "da_cuts2016 = (da_arr[\"ymd\"]>ymd_min)&(da_arr[\"ymd\"]cos_min)&(np.cos(da_arr[\"a_th\"])lam_min)\n", + " " + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "application/javascript": [ + "/* Put everything inside the global mpl namespace */\n", + "window.mpl = {};\n", + "\n", + "\n", + "mpl.get_websocket_type = function() {\n", + " if (typeof(WebSocket) !== 'undefined') {\n", + " return WebSocket;\n", + " } else if (typeof(MozWebSocket) !== 'undefined') {\n", + " return MozWebSocket;\n", + " } else {\n", + " alert('Your browser does not have WebSocket support.' +\n", + " 'Please try Chrome, Safari or Firefox ≥ 6. ' +\n", + " 'Firefox 4 and 5 are also supported but you ' +\n", + " 'have to enable WebSockets in about:config.');\n", + " };\n", + "}\n", + "\n", + "mpl.figure = function(figure_id, websocket, ondownload, parent_element) {\n", + " this.id = figure_id;\n", + "\n", + " this.ws = websocket;\n", + "\n", + " this.supports_binary = (this.ws.binaryType != undefined);\n", + "\n", + " if (!this.supports_binary) {\n", + " var warnings = document.getElementById(\"mpl-warnings\");\n", + " if (warnings) {\n", + " warnings.style.display = 'block';\n", + " warnings.textContent = (\n", + " \"This browser does not support binary websocket messages. \" +\n", + " \"Performance may be slow.\");\n", + " }\n", + " }\n", + "\n", + " this.imageObj = new Image();\n", + "\n", + " this.context = undefined;\n", + " this.message = undefined;\n", + " this.canvas = undefined;\n", + " this.rubberband_canvas = undefined;\n", + " this.rubberband_context = undefined;\n", + " this.format_dropdown = undefined;\n", + "\n", + " this.image_mode = 'full';\n", + "\n", + " this.root = $('
');\n", + " this._root_extra_style(this.root)\n", + " this.root.attr('style', 'display: inline-block');\n", + "\n", + " $(parent_element).append(this.root);\n", + "\n", + " this._init_header(this);\n", + " this._init_canvas(this);\n", + " this._init_toolbar(this);\n", + "\n", + " var fig = this;\n", + "\n", + " this.waiting = false;\n", + "\n", + " this.ws.onopen = function () {\n", + " fig.send_message(\"supports_binary\", {value: fig.supports_binary});\n", + " fig.send_message(\"send_image_mode\", {});\n", + " if (mpl.ratio != 1) {\n", + " fig.send_message(\"set_dpi_ratio\", {'dpi_ratio': mpl.ratio});\n", + " }\n", + " fig.send_message(\"refresh\", {});\n", + " }\n", + "\n", + " this.imageObj.onload = function() {\n", + " if (fig.image_mode == 'full') {\n", + " // Full images could contain transparency (where diff images\n", + " // almost always do), so we need to clear the canvas so that\n", + " // there is no ghosting.\n", + " fig.context.clearRect(0, 0, fig.canvas.width, fig.canvas.height);\n", + " }\n", + " fig.context.drawImage(fig.imageObj, 0, 0);\n", + " };\n", + "\n", + " this.imageObj.onunload = function() {\n", + " fig.ws.close();\n", + " }\n", + "\n", + " this.ws.onmessage = this._make_on_message_function(this);\n", + "\n", + " this.ondownload = ondownload;\n", + "}\n", + "\n", + "mpl.figure.prototype._init_header = function() {\n", + " var titlebar = $(\n", + " '
');\n", + " var titletext = $(\n", + " '
');\n", + " titlebar.append(titletext)\n", + " this.root.append(titlebar);\n", + " this.header = titletext[0];\n", + "}\n", + "\n", + "\n", + "\n", + "mpl.figure.prototype._canvas_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "\n", + "mpl.figure.prototype._root_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "mpl.figure.prototype._init_canvas = function() {\n", + " var fig = this;\n", + "\n", + " var canvas_div = $('
');\n", + "\n", + " canvas_div.attr('style', 'position: relative; clear: both; outline: 0');\n", + "\n", + " function canvas_keyboard_event(event) {\n", + " return fig.key_event(event, event['data']);\n", + " }\n", + "\n", + " canvas_div.keydown('key_press', canvas_keyboard_event);\n", + " canvas_div.keyup('key_release', canvas_keyboard_event);\n", + " this.canvas_div = canvas_div\n", + " this._canvas_extra_style(canvas_div)\n", + " this.root.append(canvas_div);\n", + "\n", + " var canvas = $('');\n", + " canvas.addClass('mpl-canvas');\n", + " canvas.attr('style', \"left: 0; top: 0; z-index: 0; outline: 0\")\n", + "\n", + " this.canvas = canvas[0];\n", + " this.context = canvas[0].getContext(\"2d\");\n", + "\n", + " var backingStore = this.context.backingStorePixelRatio ||\n", + "\tthis.context.webkitBackingStorePixelRatio ||\n", + "\tthis.context.mozBackingStorePixelRatio ||\n", + "\tthis.context.msBackingStorePixelRatio ||\n", + "\tthis.context.oBackingStorePixelRatio ||\n", + "\tthis.context.backingStorePixelRatio || 1;\n", + "\n", + " mpl.ratio = (window.devicePixelRatio || 1) / backingStore;\n", + "\n", + " var rubberband = $('');\n", + " rubberband.attr('style', \"position: absolute; left: 0; top: 0; z-index: 1;\")\n", + "\n", + " var pass_mouse_events = true;\n", + "\n", + " canvas_div.resizable({\n", + " start: function(event, ui) {\n", + " pass_mouse_events = false;\n", + " },\n", + " resize: function(event, ui) {\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " stop: function(event, ui) {\n", + " pass_mouse_events = true;\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " });\n", + "\n", + " function mouse_event_fn(event) {\n", + " if (pass_mouse_events)\n", + " return fig.mouse_event(event, event['data']);\n", + " }\n", + "\n", + " rubberband.mousedown('button_press', mouse_event_fn);\n", + " rubberband.mouseup('button_release', mouse_event_fn);\n", + " // Throttle sequential mouse events to 1 every 20ms.\n", + " rubberband.mousemove('motion_notify', mouse_event_fn);\n", + "\n", + " rubberband.mouseenter('figure_enter', mouse_event_fn);\n", + " rubberband.mouseleave('figure_leave', mouse_event_fn);\n", + "\n", + " canvas_div.on(\"wheel\", function (event) {\n", + " event = event.originalEvent;\n", + " event['data'] = 'scroll'\n", + " if (event.deltaY < 0) {\n", + " event.step = 1;\n", + " } else {\n", + " event.step = -1;\n", + " }\n", + " mouse_event_fn(event);\n", + " });\n", + "\n", + " canvas_div.append(canvas);\n", + " canvas_div.append(rubberband);\n", + "\n", + " this.rubberband = rubberband;\n", + " this.rubberband_canvas = rubberband[0];\n", + " this.rubberband_context = rubberband[0].getContext(\"2d\");\n", + " this.rubberband_context.strokeStyle = \"#000000\";\n", + "\n", + " this._resize_canvas = function(width, height) {\n", + " // Keep the size of the canvas, canvas container, and rubber band\n", + " // canvas in synch.\n", + " canvas_div.css('width', width)\n", + " canvas_div.css('height', height)\n", + "\n", + " canvas.attr('width', width * mpl.ratio);\n", + " canvas.attr('height', height * mpl.ratio);\n", + " canvas.attr('style', 'width: ' + width + 'px; height: ' + height + 'px;');\n", + "\n", + " rubberband.attr('width', width);\n", + " rubberband.attr('height', height);\n", + " }\n", + "\n", + " // Set the figure to an initial 600x600px, this will subsequently be updated\n", + " // upon first draw.\n", + " this._resize_canvas(600, 600);\n", + "\n", + " // Disable right mouse context menu.\n", + " $(this.rubberband_canvas).bind(\"contextmenu\",function(e){\n", + " return false;\n", + " });\n", + "\n", + " function set_focus () {\n", + " canvas.focus();\n", + " canvas_div.focus();\n", + " }\n", + "\n", + " window.setTimeout(set_focus, 100);\n", + "}\n", + "\n", + "mpl.figure.prototype._init_toolbar = function() {\n", + " var fig = this;\n", + "\n", + " var nav_element = $('
')\n", + " nav_element.attr('style', 'width: 100%');\n", + " this.root.append(nav_element);\n", + "\n", + " // Define a callback function for later on.\n", + " function toolbar_event(event) {\n", + " return fig.toolbar_button_onclick(event['data']);\n", + " }\n", + " function toolbar_mouse_event(event) {\n", + " return fig.toolbar_button_onmouseover(event['data']);\n", + " }\n", + "\n", + " for(var toolbar_ind in mpl.toolbar_items) {\n", + " var name = mpl.toolbar_items[toolbar_ind][0];\n", + " var tooltip = mpl.toolbar_items[toolbar_ind][1];\n", + " var image = mpl.toolbar_items[toolbar_ind][2];\n", + " var method_name = mpl.toolbar_items[toolbar_ind][3];\n", + "\n", + " if (!name) {\n", + " // put a spacer in here.\n", + " continue;\n", + " }\n", + " var button = $('');\n", + " button.click(method_name, toolbar_event);\n", + " button.mouseover(tooltip, toolbar_mouse_event);\n", + " nav_element.append(button);\n", + " }\n", + "\n", + " // Add the status bar.\n", + " var status_bar = $('');\n", + " nav_element.append(status_bar);\n", + " this.message = status_bar[0];\n", + "\n", + " // Add the close button to the window.\n", + " var buttongrp = $('
');\n", + " var button = $('');\n", + " button.click(function (evt) { fig.handle_close(fig, {}); } );\n", + " button.mouseover('Stop Interaction', toolbar_mouse_event);\n", + " buttongrp.append(button);\n", + " var titlebar = this.root.find($('.ui-dialog-titlebar'));\n", + " titlebar.prepend(buttongrp);\n", + "}\n", + "\n", + "mpl.figure.prototype._root_extra_style = function(el){\n", + " var fig = this\n", + " el.on(\"remove\", function(){\n", + "\tfig.close_ws(fig, {});\n", + " });\n", + "}\n", + "\n", + "mpl.figure.prototype._canvas_extra_style = function(el){\n", + " // this is important to make the div 'focusable\n", + " el.attr('tabindex', 0)\n", + " // reach out to IPython and tell the keyboard manager to turn it's self\n", + " // off when our div gets focus\n", + "\n", + " // location in version 3\n", + " if (IPython.notebook.keyboard_manager) {\n", + " IPython.notebook.keyboard_manager.register_events(el);\n", + " }\n", + " else {\n", + " // location in version 2\n", + " IPython.keyboard_manager.register_events(el);\n", + " }\n", + "\n", + "}\n", + "\n", + "mpl.figure.prototype._key_event_extra = function(event, name) {\n", + " var manager = IPython.notebook.keyboard_manager;\n", + " if (!manager)\n", + " manager = IPython.keyboard_manager;\n", + "\n", + " // Check for shift+enter\n", + " if (event.shiftKey && event.which == 13) {\n", + " this.canvas_div.blur();\n", + " event.shiftKey = false;\n", + " // Send a \"J\" for go to next cell\n", + " event.which = 74;\n", + " event.keyCode = 74;\n", + " manager.command_mode();\n", + " manager.handle_keydown(event);\n", + " }\n", + "}\n", + "\n", + "mpl.figure.prototype.handle_save = function(fig, msg) {\n", + " fig.ondownload(fig, null);\n", + "}\n", + "\n", + "\n", + "mpl.find_output_cell = function(html_output) {\n", + " // Return the cell and output element which can be found *uniquely* in the notebook.\n", + " // Note - this is a bit hacky, but it is done because the \"notebook_saving.Notebook\"\n", + " // IPython event is triggered only after the cells have been serialised, which for\n", + " // our purposes (turning an active figure into a static one), is too late.\n", + " var cells = IPython.notebook.get_cells();\n", + " var ncells = cells.length;\n", + " for (var i=0; i= 3 moved mimebundle to data attribute of output\n", + " data = data.data;\n", + " }\n", + " if (data['text/html'] == html_output) {\n", + " return [cell, data, j];\n", + " }\n", + " }\n", + " }\n", + " }\n", + "}\n", + "\n", + "// Register the function which deals with the matplotlib target/channel.\n", + "// The kernel may be null if the page has been refreshed.\n", + "if (IPython.notebook.kernel != null) {\n", + " IPython.notebook.kernel.comm_manager.register_target('matplotlib', mpl.mpl_figure_comm);\n", + "}\n" + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + }, + { + "data": { + "text/html": [ + "" + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + } + ], + "source": [ + "bins = np.linspace(0,2,64)\n", + "fonto = 30\n", + "\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_beta\"])[mu_cuts2016],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2016])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_beta\"])[nu_cuts2016],bins=bins,weights=w_bg_nu[nu_cuts2016])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_beta\"])[anu_cuts2016],bins=bins,weights=w_bg_anu[anu_cuts2016])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_beta\"])[da_cuts2016],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "fig, ax = plt.subplots(2,2,figsize=[16,16])\n", + "fig.text(.46,.5,\"2007-2016\",fontsize=30)\n", + "ax[0,0].set_xlabel(r\"$\\beta$\",fontsize=fonto)\n", + "ax[0,0].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[0,0].set_yscale(\"log\")\n", + "ax[0,0].set_xlim([0,1.4])\n", + "ax[0,0].set_ylim([1e-3,1e5])\n", + "ax[0,0].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[0,0].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[0,0].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[0,0].scatter(xda_c,bg_da,marker=\"+\",color='k',s=70,lw=1.5,label=\"data\")\n", + "\n", + "legend = ax[0,0].legend(loc='lower right', ncol=1, shadow=False, fontsize=15)\n", + "ax[0,0].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[0,0].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "bins = np.linspace(-6,-3,64)\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_lambda\"])[mu_cuts2016],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2016])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_lambda\"])[nu_cuts2016],bins=bins,weights=w_bg_nu[nu_cuts2016])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_lambda\"])[anu_cuts2016],bins=bins,weights=w_bg_anu[anu_cuts2016])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_lambda\"])[da_cuts2016],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "#fig, ax = plt.subplots(figsize=[12,7])\n", + "ax[1,0].set_xlabel(r\"$\\lambda$\",fontsize=fonto)\n", + "ax[1,0].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[1,0].set_yscale(\"log\")\n", + "ax[1,0].set_xlim([-6,-3])\n", + "ax[1,0].set_ylim([1e-3,1e6])\n", + "ax[1,0].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[1,0].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[1,0].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[1,0].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "\n", + "legend = ax[1,0].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[1,0].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[1,0].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "\n", + "bins = np.linspace(0,1,64)\n", + "\n", + "bg_mu,x_mu = np.histogram(np.cos((mu_arr[\"a_th\"])[mu_cuts2016]),bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2016])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram(np.cos((nu_arr[\"a_th\"])[nu_cuts2016]),bins=bins,weights=w_bg_nu[nu_cuts2016])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((np.cos(anu_arr[\"a_th\"])[anu_cuts2016]),bins=bins,weights=w_bg_anu[anu_cuts2016])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((np.cos(da_arr[\"a_th\"])[da_cuts2016]),bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "\n", + "#fig, ax1 = plt.subplots(figsize=[7,7])\n", + "\n", + "ax[0,1].set_xlabel(r\"$\\cos(\\theta)$\",fontsize=fonto)\n", + "ax[0,1].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[0,1].set_yscale(\"log\")\n", + "ax[0,1].set_xlim([0,1])\n", + "ax[0,1].set_ylim([10,30000])\n", + "\n", + "ax[0,1].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[0,1].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[0,1].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[0,1].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "legend = ax[0,1].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[0,1].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[0,1].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "\n", + "bins = np.linspace(0,120,120)\n", + "\n", + "\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_nhits\"])[mu_cuts2016],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2016])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_nhits\"])[nu_cuts2016],bins=bins,weights=w_bg_nu[nu_cuts2016])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_nhits\"])[anu_cuts2016],bins=bins,weights=w_bg_anu[anu_cuts2016])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_nhits\"])[da_cuts2016],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "#fig, ax = plt.subplots(figsize=[7,7])\n", + "ax[1,1].set_xlabel(r\"$N_{hits}$\",fontsize=fonto)\n", + "ax[1,1].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[1,1].set_yscale(\"log\")\n", + "ax[1,1].set_ylim([1,40000])\n", + "ax[1,1].set_xlim([0,100])\n", + "\n", + "ax[1,1].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[1,1].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[1,1].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[1,1].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "legend = ax[1,1].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[1,1].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[1,1].axvline(x=-5.4,linestyle='--',color='gray')\n", + "ax[0,0].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[0,0].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[0,1].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[0,1].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[1,1].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[1,1].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[1,0].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[1,0].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "\n", + "\n", + "fig.tight_layout()" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [] + }, + { + "cell_type": "code", + "execution_count": 10, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "\n", + "w_sg_nu_tt = nu_arr[\"mc_rbr\"]*nu_arr[\"mc_wc\"]*nu_arr[\"mc_w2\"]*nu_w_tt_2000\n", + "w_sg_nu_WW = nu_arr[\"mc_rbr\"]*nu_arr[\"mc_wc\"]*nu_arr[\"mc_w2\"]*nu_w_WW_2000\n", + "w_sg_nu_bb = nu_arr[\"mc_rbr\"]*nu_arr[\"mc_wc\"]*nu_arr[\"mc_w2\"]*nu_w_bb_2000\n", + "w_sg_anu_tt = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*anu_arr[\"mc_w2\"]*anu_w_tt_2000\n", + "w_sg_anu_WW = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*anu_arr[\"mc_w2\"]*anu_w_WW_2000\n", + "w_sg_anu_bb = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*anu_arr[\"mc_w2\"]*anu_w_bb_2000\n", + "\n", + "w_sg_nu_tt100 = nu_arr[\"mc_rbr\"]*nu_arr[\"mc_wc\"]*nu_arr[\"mc_w2\"]*nu_w_tt_100\n", + "w_sg_nu_bb100 = nu_arr[\"mc_rbr\"]*nu_arr[\"mc_wc\"]*nu_arr[\"mc_w2\"]*nu_w_bb_100\n", + "w_sg_anu_tt100 = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*anu_arr[\"mc_w2\"]*anu_w_tt_100\n", + "w_sg_anu_bb100 = anu_arr[\"mc_rbr\"]*anu_arr[\"mc_wc\"]*anu_arr[\"mc_w2\"]*anu_w_bb_100\n", + "\n", + "\n", + "bins = np.linspace(0,2000,60)\n", + "dowenorm = False\n", + "sg_tt,x_sg_tt = np.histogram(nu_arr[\"mc_ene\"],bins=bins,weights=w_sg_nu_tt,normed=dowenorm)\n", + "xsgc_tt = (x_sg_tt[:-1]+x_sg_tt[1:])/2\n", + "\n", + "sg_WW,x_sg_WW = np.histogram(nu_arr[\"mc_ene\"],bins=bins,weights=w_sg_nu_WW,normed=dowenorm)\n", + "xsgc_WW = (x_sg_tt[:-1]+x_sg_WW[1:])/2\n", + "\n", + "sg_bb,x_sg_bb = np.histogram(nu_arr[\"mc_ene\"],bins=bins,weights=w_sg_nu_bb,normed=dowenorm)\n", + "xsgc_bb = (x_sg_bb[:-1]+x_sg_bb[1:])/2\n", + "\n" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "x_spec = np.linspace(0.0025,1.,200)*188\n", + "\n", + "bins = np.linspace(0,200,60)\n", + "dowenorm = False\n", + "sg_tt,x_sg_tt = np.histogram(nu_arr[\"a_nhits\"],bins=bins,weights=w_sg_nu_tt,normed=dowenorm)\n", + "xsgc_tt = (x_sg_tt[:-1]+x_sg_tt[1:])/2\n", + "\n", + "sg_bb,x_sg_bb = np.histogram(nu_arr[\"a_nhits\"],bins=bins,weights=w_sg_nu_bb,normed=dowenorm)\n", + "xsgc_bb = (x_sg_bb[:-1]+x_sg_bb[1:])/2\n", + "\n", + "sg_tt100,x_sg_tt = np.histogram(nu_arr[\"a_nhits\"],bins=bins,weights=w_sg_nu_tt100,normed=dowenorm)\n", + "xsgc_tt100 = (x_sg_tt[:-1]+x_sg_tt[1:])/2\n", + "\n", + "sg_bb100,x_sg_bb = np.histogram(nu_arr[\"a_nhits\"],bins=bins,weights=w_sg_nu_bb100,normed=dowenorm)\n", + "xsgc_bb100 = (x_sg_bb[:-1]+x_sg_bb[1:])/2\n" + ] + }, + { + "cell_type": "code", + "execution_count": 12, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "data": { + "application/javascript": [ + "/* Put everything inside the global mpl namespace */\n", + "window.mpl = {};\n", + "\n", + "\n", + "mpl.get_websocket_type = function() {\n", + " if (typeof(WebSocket) !== 'undefined') {\n", + " return WebSocket;\n", + " } else if (typeof(MozWebSocket) !== 'undefined') {\n", + " return MozWebSocket;\n", + " } else {\n", + " alert('Your browser does not have WebSocket support.' +\n", + " 'Please try Chrome, Safari or Firefox ≥ 6. ' +\n", + " 'Firefox 4 and 5 are also supported but you ' +\n", + " 'have to enable WebSockets in about:config.');\n", + " };\n", + "}\n", + "\n", + "mpl.figure = function(figure_id, websocket, ondownload, parent_element) {\n", + " this.id = figure_id;\n", + "\n", + " this.ws = websocket;\n", + "\n", + " this.supports_binary = (this.ws.binaryType != undefined);\n", + "\n", + " if (!this.supports_binary) {\n", + " var warnings = document.getElementById(\"mpl-warnings\");\n", + " if (warnings) {\n", + " warnings.style.display = 'block';\n", + " warnings.textContent = (\n", + " \"This browser does not support binary websocket messages. \" +\n", + " \"Performance may be slow.\");\n", + " }\n", + " }\n", + "\n", + " this.imageObj = new Image();\n", + "\n", + " this.context = undefined;\n", + " this.message = undefined;\n", + " this.canvas = undefined;\n", + " this.rubberband_canvas = undefined;\n", + " this.rubberband_context = undefined;\n", + " this.format_dropdown = undefined;\n", + "\n", + " this.image_mode = 'full';\n", + "\n", + " this.root = $('
');\n", + " this._root_extra_style(this.root)\n", + " this.root.attr('style', 'display: inline-block');\n", + "\n", + " $(parent_element).append(this.root);\n", + "\n", + " this._init_header(this);\n", + " this._init_canvas(this);\n", + " this._init_toolbar(this);\n", + "\n", + " var fig = this;\n", + "\n", + " this.waiting = false;\n", + "\n", + " this.ws.onopen = function () {\n", + " fig.send_message(\"supports_binary\", {value: fig.supports_binary});\n", + " fig.send_message(\"send_image_mode\", {});\n", + " if (mpl.ratio != 1) {\n", + " fig.send_message(\"set_dpi_ratio\", {'dpi_ratio': mpl.ratio});\n", + " }\n", + " fig.send_message(\"refresh\", {});\n", + " }\n", + "\n", + " this.imageObj.onload = function() {\n", + " if (fig.image_mode == 'full') {\n", + " // Full images could contain transparency (where diff images\n", + " // almost always do), so we need to clear the canvas so that\n", + " // there is no ghosting.\n", + " fig.context.clearRect(0, 0, fig.canvas.width, fig.canvas.height);\n", + " }\n", + " fig.context.drawImage(fig.imageObj, 0, 0);\n", + " };\n", + "\n", + " this.imageObj.onunload = function() {\n", + " fig.ws.close();\n", + " }\n", + "\n", + " this.ws.onmessage = this._make_on_message_function(this);\n", + "\n", + " this.ondownload = ondownload;\n", + "}\n", + "\n", + "mpl.figure.prototype._init_header = function() {\n", + " var titlebar = $(\n", + " '
');\n", + " var titletext = $(\n", + " '
');\n", + " titlebar.append(titletext)\n", + " this.root.append(titlebar);\n", + " this.header = titletext[0];\n", + "}\n", + "\n", + "\n", + "\n", + "mpl.figure.prototype._canvas_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "\n", + "mpl.figure.prototype._root_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "mpl.figure.prototype._init_canvas = function() {\n", + " var fig = this;\n", + "\n", + " var canvas_div = $('
');\n", + "\n", + " canvas_div.attr('style', 'position: relative; clear: both; outline: 0');\n", + "\n", + " function canvas_keyboard_event(event) {\n", + " return fig.key_event(event, event['data']);\n", + " }\n", + "\n", + " canvas_div.keydown('key_press', canvas_keyboard_event);\n", + " canvas_div.keyup('key_release', canvas_keyboard_event);\n", + " this.canvas_div = canvas_div\n", + " this._canvas_extra_style(canvas_div)\n", + " this.root.append(canvas_div);\n", + "\n", + " var canvas = $('');\n", + " canvas.addClass('mpl-canvas');\n", + " canvas.attr('style', \"left: 0; top: 0; z-index: 0; outline: 0\")\n", + "\n", + " this.canvas = canvas[0];\n", + " this.context = canvas[0].getContext(\"2d\");\n", + "\n", + " var backingStore = this.context.backingStorePixelRatio ||\n", + "\tthis.context.webkitBackingStorePixelRatio ||\n", + "\tthis.context.mozBackingStorePixelRatio ||\n", + "\tthis.context.msBackingStorePixelRatio ||\n", + "\tthis.context.oBackingStorePixelRatio ||\n", + "\tthis.context.backingStorePixelRatio || 1;\n", + "\n", + " mpl.ratio = (window.devicePixelRatio || 1) / backingStore;\n", + "\n", + " var rubberband = $('');\n", + " rubberband.attr('style', \"position: absolute; left: 0; top: 0; z-index: 1;\")\n", + "\n", + " var pass_mouse_events = true;\n", + "\n", + " canvas_div.resizable({\n", + " start: function(event, ui) {\n", + " pass_mouse_events = false;\n", + " },\n", + " resize: function(event, ui) {\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " stop: function(event, ui) {\n", + " pass_mouse_events = true;\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " });\n", + "\n", + " function mouse_event_fn(event) {\n", + " if (pass_mouse_events)\n", + " return fig.mouse_event(event, event['data']);\n", + " }\n", + "\n", + " rubberband.mousedown('button_press', mouse_event_fn);\n", + " rubberband.mouseup('button_release', mouse_event_fn);\n", + " // Throttle sequential mouse events to 1 every 20ms.\n", + " rubberband.mousemove('motion_notify', mouse_event_fn);\n", + "\n", + " rubberband.mouseenter('figure_enter', mouse_event_fn);\n", + " rubberband.mouseleave('figure_leave', mouse_event_fn);\n", + "\n", + " canvas_div.on(\"wheel\", function (event) {\n", + " event = event.originalEvent;\n", + " event['data'] = 'scroll'\n", + " if (event.deltaY < 0) {\n", + " event.step = 1;\n", + " } else {\n", + " event.step = -1;\n", + " }\n", + " mouse_event_fn(event);\n", + " });\n", + "\n", + " canvas_div.append(canvas);\n", + " canvas_div.append(rubberband);\n", + "\n", + " this.rubberband = rubberband;\n", + " this.rubberband_canvas = rubberband[0];\n", + " this.rubberband_context = rubberband[0].getContext(\"2d\");\n", + " this.rubberband_context.strokeStyle = \"#000000\";\n", + "\n", + " this._resize_canvas = function(width, height) {\n", + " // Keep the size of the canvas, canvas container, and rubber band\n", + " // canvas in synch.\n", + " canvas_div.css('width', width)\n", + " canvas_div.css('height', height)\n", + "\n", + " canvas.attr('width', width * mpl.ratio);\n", + " canvas.attr('height', height * mpl.ratio);\n", + " canvas.attr('style', 'width: ' + width + 'px; height: ' + height + 'px;');\n", + "\n", + " rubberband.attr('width', width);\n", + " rubberband.attr('height', height);\n", + " }\n", + "\n", + " // Set the figure to an initial 600x600px, this will subsequently be updated\n", + " // upon first draw.\n", + " this._resize_canvas(600, 600);\n", + "\n", + " // Disable right mouse context menu.\n", + " $(this.rubberband_canvas).bind(\"contextmenu\",function(e){\n", + " return false;\n", + " });\n", + "\n", + " function set_focus () {\n", + " canvas.focus();\n", + " canvas_div.focus();\n", + " }\n", + "\n", + " window.setTimeout(set_focus, 100);\n", + "}\n", + "\n", + "mpl.figure.prototype._init_toolbar = function() {\n", + " var fig = this;\n", + "\n", + " var nav_element = $('
')\n", + " nav_element.attr('style', 'width: 100%');\n", + " this.root.append(nav_element);\n", + "\n", + " // Define a callback function for later on.\n", + " function toolbar_event(event) {\n", + " return fig.toolbar_button_onclick(event['data']);\n", + " }\n", + " function toolbar_mouse_event(event) {\n", + " return fig.toolbar_button_onmouseover(event['data']);\n", + " }\n", + "\n", + " for(var toolbar_ind in mpl.toolbar_items) {\n", + " var name = mpl.toolbar_items[toolbar_ind][0];\n", + " var tooltip = mpl.toolbar_items[toolbar_ind][1];\n", + " var image = mpl.toolbar_items[toolbar_ind][2];\n", + " var method_name = mpl.toolbar_items[toolbar_ind][3];\n", + "\n", + " if (!name) {\n", + " // put a spacer in here.\n", + " continue;\n", + " }\n", + " var button = $('');\n", + " button.click(method_name, toolbar_event);\n", + " button.mouseover(tooltip, toolbar_mouse_event);\n", + " nav_element.append(button);\n", + " }\n", + "\n", + " // Add the status bar.\n", + " var status_bar = $('');\n", + " nav_element.append(status_bar);\n", + " this.message = status_bar[0];\n", + "\n", + " // Add the close button to the window.\n", + " var buttongrp = $('
');\n", + " var button = $('');\n", + " button.click(function (evt) { fig.handle_close(fig, {}); } );\n", + " button.mouseover('Stop Interaction', toolbar_mouse_event);\n", + " buttongrp.append(button);\n", + " var titlebar = this.root.find($('.ui-dialog-titlebar'));\n", + " titlebar.prepend(buttongrp);\n", + "}\n", + "\n", + "mpl.figure.prototype._root_extra_style = function(el){\n", + " var fig = this\n", + " el.on(\"remove\", function(){\n", + "\tfig.close_ws(fig, {});\n", + " });\n", + "}\n", + "\n", + "mpl.figure.prototype._canvas_extra_style = function(el){\n", + " // this is important to make the div 'focusable\n", + " el.attr('tabindex', 0)\n", + " // reach out to IPython and tell the keyboard manager to turn it's self\n", + " // off when our div gets focus\n", + "\n", + " // location in version 3\n", + " if (IPython.notebook.keyboard_manager) {\n", + " IPython.notebook.keyboard_manager.register_events(el);\n", + " }\n", + " else {\n", + " // location in version 2\n", + " IPython.keyboard_manager.register_events(el);\n", + " }\n", + "\n", + "}\n", + "\n", + "mpl.figure.prototype._key_event_extra = function(event, name) {\n", + " var manager = IPython.notebook.keyboard_manager;\n", + " if (!manager)\n", + " manager = IPython.keyboard_manager;\n", + "\n", + " // Check for shift+enter\n", + " if (event.shiftKey && event.which == 13) {\n", + " this.canvas_div.blur();\n", + " event.shiftKey = false;\n", + " // Send a \"J\" for go to next cell\n", + " event.which = 74;\n", + " event.keyCode = 74;\n", + " manager.command_mode();\n", + " manager.handle_keydown(event);\n", + " }\n", + "}\n", + "\n", + "mpl.figure.prototype.handle_save = function(fig, msg) {\n", + " fig.ondownload(fig, null);\n", + "}\n", + "\n", + "\n", + "mpl.find_output_cell = function(html_output) {\n", + " // Return the cell and output element which can be found *uniquely* in the notebook.\n", + " // Note - this is a bit hacky, but it is done because the \"notebook_saving.Notebook\"\n", + " // IPython event is triggered only after the cells have been serialised, which for\n", + " // our purposes (turning an active figure into a static one), is too late.\n", + " var cells = IPython.notebook.get_cells();\n", + " var ncells = cells.length;\n", + " for (var i=0; i= 3 moved mimebundle to data attribute of output\n", + " data = data.data;\n", + " }\n", + " if (data['text/html'] == html_output) {\n", + " return [cell, data, j];\n", + " }\n", + " }\n", + " }\n", + " }\n", + "}\n", + "\n", + "// Register the function which deals with the matplotlib target/channel.\n", + "// The kernel may be null if the page has been refreshed.\n", + "if (IPython.notebook.kernel != null) {\n", + " IPython.notebook.kernel.comm_manager.register_target('matplotlib', mpl.mpl_figure_comm);\n", + "}\n" + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + }, + { + "data": { + "text/html": [ + "" + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + } + ], + "source": [ + "bins = np.linspace(0,2,64)\n", + "fonto = 30\n", + "\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_beta\"])[mu_cuts2012],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2012])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_beta\"])[nu_cuts2012],bins=bins,weights=w_bg_nu[nu_cuts2012])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_beta\"])[anu_cuts2012],bins=bins,weights=w_bg_anu[anu_cuts2012])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_beta\"])[da_cuts2012],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "fig, ax = plt.subplots(2,2,figsize=[16,16])\n", + "fig.text(.46,.5,\"2007-2012\",fontsize=30)\n", + "ax[0,0].set_xlabel(r\"$\\beta$\",fontsize=fonto)\n", + "ax[0,0].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[0,0].set_yscale(\"log\")\n", + "ax[0,0].set_xlim([0,1.4])\n", + "ax[0,0].set_ylim([1e-3,1e5])\n", + "ax[0,0].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[0,0].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[0,0].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[0,0].scatter(xda_c,bg_da,marker=\"+\",color='k',s=70,lw=1.5,label=\"data\")\n", + "\n", + "legend = ax[0,0].legend(loc='lower right', ncol=1, shadow=False, fontsize=15)\n", + "ax[0,0].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[0,0].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "bins = np.linspace(-6,-3,64)\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_lambda\"])[mu_cuts2012],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2012])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_lambda\"])[nu_cuts2012],bins=bins,weights=w_bg_nu[nu_cuts2012])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_lambda\"])[anu_cuts2012],bins=bins,weights=w_bg_anu[anu_cuts2012])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_lambda\"])[da_cuts2012],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "#fig, ax = plt.subplots(figsize=[12,7])\n", + "ax[1,0].set_xlabel(r\"$\\lambda$\",fontsize=fonto)\n", + "ax[1,0].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[1,0].set_yscale(\"log\")\n", + "ax[1,0].set_xlim([-6,-3])\n", + "ax[1,0].set_ylim([1e-3,1e6])\n", + "ax[1,0].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[1,0].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[1,0].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[1,0].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "\n", + "legend = ax[1,0].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[1,0].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[1,0].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "\n", + "bins = np.linspace(0,1,64)\n", + "\n", + "bg_mu,x_mu = np.histogram(np.cos((mu_arr[\"a_th\"])[mu_cuts2012]),bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2012])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram(np.cos((nu_arr[\"a_th\"])[nu_cuts2012]),bins=bins,weights=w_bg_nu[nu_cuts2012])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((np.cos(anu_arr[\"a_th\"])[anu_cuts2012]),bins=bins,weights=w_bg_anu[anu_cuts2012])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((np.cos(da_arr[\"a_th\"])[da_cuts2012]),bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "\n", + "#fig, ax1 = plt.subplots(figsize=[7,7])\n", + "\n", + "ax[0,1].set_xlabel(r\"$\\cos(\\theta)$\",fontsize=fonto)\n", + "ax[0,1].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[0,1].set_yscale(\"log\")\n", + "ax[0,1].set_xlim([0,1])\n", + "ax[0,1].set_ylim([10,30000])\n", + "\n", + "ax[0,1].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[0,1].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[0,1].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[0,1].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "legend = ax[0,1].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[0,1].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[0,1].axvline(x=-5.4,linestyle='--',color='gray')\n", + "\n", + "fig.tight_layout()\n", + "\n", + "\n", + "bins = np.linspace(0,120,120)\n", + "\n", + "\n", + "bg_mu,x_mu = np.histogram((mu_arr[\"a_nhits\"])[mu_cuts2012],bins=bins,weights=(mu_arr[\"mc_wc\"])[mu_cuts2012])\n", + "xmu_c = (x_mu[:-1]+x_mu[1:])/2\n", + "\n", + "bg_nu,x_bg = np.histogram((nu_arr[\"a_nhits\"])[nu_cuts2012],bins=bins,weights=w_bg_nu[nu_cuts2012])\n", + "xnu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "bg_anu,x_abg = np.histogram((anu_arr[\"a_nhits\"])[anu_cuts2012],bins=bins,weights=w_bg_anu[anu_cuts2012])\n", + "xanu_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "bg_da,x_abg = np.histogram((da_arr[\"a_nhits\"])[da_cuts2012],bins=bins)\n", + "xda_c = (x_bg[:-1]+x_bg[1:])/2\n", + "\n", + "#fig, ax = plt.subplots(figsize=[7,7])\n", + "ax[1,1].set_xlabel(r\"$N_{hits}$\",fontsize=fonto)\n", + "ax[1,1].set_ylabel(r\"Number of events\",fontsize=20)\n", + "\n", + "ax[1,1].set_yscale(\"log\")\n", + "ax[1,1].set_ylim([1,40000])\n", + "ax[1,1].set_xlim([0,100])\n", + "\n", + "ax[1,1].step(xmu_c,bg_mu,where=\"mid\",lw=2,label=r\"$\\mu$\")\n", + "ax[1,1].step(xnu_c,bg_nu+bg_anu,where=\"mid\",lw=2,label=r\"$\\nu + \\bar{\\nu}$\")\n", + "\n", + "ax[1,1].step(xnu_c,bg_nu+bg_anu+bg_mu,where=\"mid\",linestyle=\"--\",lw=2,label=\"total\")\n", + "ax[1,1].scatter(xda_c,bg_da,marker=\"+\",color='k',lw=1.5,s=70,label=\"data\")\n", + "legend = ax[1,1].legend(loc='upper right', ncol=1, shadow=False, fontsize=15)\n", + "ax[1,1].axvline(x=-5.2,linestyle='--',color='k')\n", + "ax[1,1].axvline(x=-5.4,linestyle='--',color='gray')\n", + "ax[0,0].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[0,0].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[0,1].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[0,1].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[1,1].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[1,1].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "ax[1,0].tick_params(axis='both', which='major', labelsize=15, size=5,width=1.2)\n", + "ax[1,0].tick_params(axis='both', which='minor', labelsize=15, size=3,width=1.2)\n", + "\n", + "\n", + "fig.tight_layout()" + ] + }, + { + "cell_type": "code", + "execution_count": 15, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "application/javascript": [ + "/* Put everything inside the global mpl namespace */\n", + "window.mpl = {};\n", + "\n", + "\n", + "mpl.get_websocket_type = function() {\n", + " if (typeof(WebSocket) !== 'undefined') {\n", + " return WebSocket;\n", + " } else if (typeof(MozWebSocket) !== 'undefined') {\n", + " return MozWebSocket;\n", + " } else {\n", + " alert('Your browser does not have WebSocket support.' +\n", + " 'Please try Chrome, Safari or Firefox ≥ 6. ' +\n", + " 'Firefox 4 and 5 are also supported but you ' +\n", + " 'have to enable WebSockets in about:config.');\n", + " };\n", + "}\n", + "\n", + "mpl.figure = function(figure_id, websocket, ondownload, parent_element) {\n", + " this.id = figure_id;\n", + "\n", + " this.ws = websocket;\n", + "\n", + " this.supports_binary = (this.ws.binaryType != undefined);\n", + "\n", + " if (!this.supports_binary) {\n", + " var warnings = document.getElementById(\"mpl-warnings\");\n", + " if (warnings) {\n", + " warnings.style.display = 'block';\n", + " warnings.textContent = (\n", + " \"This browser does not support binary websocket messages. \" +\n", + " \"Performance may be slow.\");\n", + " }\n", + " }\n", + "\n", + " this.imageObj = new Image();\n", + "\n", + " this.context = undefined;\n", + " this.message = undefined;\n", + " this.canvas = undefined;\n", + " this.rubberband_canvas = undefined;\n", + " this.rubberband_context = undefined;\n", + " this.format_dropdown = undefined;\n", + "\n", + " this.image_mode = 'full';\n", + "\n", + " this.root = $('
');\n", + " this._root_extra_style(this.root)\n", + " this.root.attr('style', 'display: inline-block');\n", + "\n", + " $(parent_element).append(this.root);\n", + "\n", + " this._init_header(this);\n", + " this._init_canvas(this);\n", + " this._init_toolbar(this);\n", + "\n", + " var fig = this;\n", + "\n", + " this.waiting = false;\n", + "\n", + " this.ws.onopen = function () {\n", + " fig.send_message(\"supports_binary\", {value: fig.supports_binary});\n", + " fig.send_message(\"send_image_mode\", {});\n", + " if (mpl.ratio != 1) {\n", + " fig.send_message(\"set_dpi_ratio\", {'dpi_ratio': mpl.ratio});\n", + " }\n", + " fig.send_message(\"refresh\", {});\n", + " }\n", + "\n", + " this.imageObj.onload = function() {\n", + " if (fig.image_mode == 'full') {\n", + " // Full images could contain transparency (where diff images\n", + " // almost always do), so we need to clear the canvas so that\n", + " // there is no ghosting.\n", + " fig.context.clearRect(0, 0, fig.canvas.width, fig.canvas.height);\n", + " }\n", + " fig.context.drawImage(fig.imageObj, 0, 0);\n", + " };\n", + "\n", + " this.imageObj.onunload = function() {\n", + " fig.ws.close();\n", + " }\n", + "\n", + " this.ws.onmessage = this._make_on_message_function(this);\n", + "\n", + " this.ondownload = ondownload;\n", + "}\n", + "\n", + "mpl.figure.prototype._init_header = function() {\n", + " var titlebar = $(\n", + " '
');\n", + " var titletext = $(\n", + " '
');\n", + " titlebar.append(titletext)\n", + " this.root.append(titlebar);\n", + " this.header = titletext[0];\n", + "}\n", + "\n", + "\n", + "\n", + "mpl.figure.prototype._canvas_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "\n", + "mpl.figure.prototype._root_extra_style = function(canvas_div) {\n", + "\n", + "}\n", + "\n", + "mpl.figure.prototype._init_canvas = function() {\n", + " var fig = this;\n", + "\n", + " var canvas_div = $('
');\n", + "\n", + " canvas_div.attr('style', 'position: relative; clear: both; outline: 0');\n", + "\n", + " function canvas_keyboard_event(event) {\n", + " return fig.key_event(event, event['data']);\n", + " }\n", + "\n", + " canvas_div.keydown('key_press', canvas_keyboard_event);\n", + " canvas_div.keyup('key_release', canvas_keyboard_event);\n", + " this.canvas_div = canvas_div\n", + " this._canvas_extra_style(canvas_div)\n", + " this.root.append(canvas_div);\n", + "\n", + " var canvas = $('');\n", + " canvas.addClass('mpl-canvas');\n", + " canvas.attr('style', \"left: 0; top: 0; z-index: 0; outline: 0\")\n", + "\n", + " this.canvas = canvas[0];\n", + " this.context = canvas[0].getContext(\"2d\");\n", + "\n", + " var backingStore = this.context.backingStorePixelRatio ||\n", + "\tthis.context.webkitBackingStorePixelRatio ||\n", + "\tthis.context.mozBackingStorePixelRatio ||\n", + "\tthis.context.msBackingStorePixelRatio ||\n", + "\tthis.context.oBackingStorePixelRatio ||\n", + "\tthis.context.backingStorePixelRatio || 1;\n", + "\n", + " mpl.ratio = (window.devicePixelRatio || 1) / backingStore;\n", + "\n", + " var rubberband = $('');\n", + " rubberband.attr('style', \"position: absolute; left: 0; top: 0; z-index: 1;\")\n", + "\n", + " var pass_mouse_events = true;\n", + "\n", + " canvas_div.resizable({\n", + " start: function(event, ui) {\n", + " pass_mouse_events = false;\n", + " },\n", + " resize: function(event, ui) {\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " stop: function(event, ui) {\n", + " pass_mouse_events = true;\n", + " fig.request_resize(ui.size.width, ui.size.height);\n", + " },\n", + " });\n", + "\n", + " function mouse_event_fn(event) {\n", + " if (pass_mouse_events)\n", + " return fig.mouse_event(event, event['data']);\n", + " }\n", + "\n", + " rubberband.mousedown('button_press', mouse_event_fn);\n", + " rubberband.mouseup('button_release', mouse_event_fn);\n", + " // Throttle sequential mouse events to 1 every 20ms.\n", + " rubberband.mousemove('motion_notify', mouse_event_fn);\n", + "\n", + " rubberband.mouseenter('figure_enter', mouse_event_fn);\n", + " rubberband.mouseleave('figure_leave', mouse_event_fn);\n", + "\n", + " canvas_div.on(\"wheel\", function (event) {\n", + " event = event.originalEvent;\n", + " event['data'] = 'scroll'\n", + " if (event.deltaY < 0) {\n", + " event.step = 1;\n", + " } else {\n", + " event.step = -1;\n", + " }\n", + " mouse_event_fn(event);\n", + " });\n", + "\n", + " canvas_div.append(canvas);\n", + " canvas_div.append(rubberband);\n", + "\n", + " this.rubberband = rubberband;\n", + " this.rubberband_canvas = rubberband[0];\n", + " this.rubberband_context = rubberband[0].getContext(\"2d\");\n", + " this.rubberband_context.strokeStyle = \"#000000\";\n", + "\n", + " this._resize_canvas = function(width, height) {\n", + " // Keep the size of the canvas, canvas container, and rubber band\n", + " // canvas in synch.\n", + " canvas_div.css('width', width)\n", + " canvas_div.css('height', height)\n", + "\n", + " canvas.attr('width', width * mpl.ratio);\n", + " canvas.attr('height', height * mpl.ratio);\n", + " canvas.attr('style', 'width: ' + width + 'px; height: ' + height + 'px;');\n", + "\n", + " rubberband.attr('width', width);\n", + " rubberband.attr('height', height);\n", + " }\n", + "\n", + " // Set the figure to an initial 600x600px, this will subsequently be updated\n", + " // upon first draw.\n", + " this._resize_canvas(600, 600);\n", + "\n", + " // Disable right mouse context menu.\n", + " $(this.rubberband_canvas).bind(\"contextmenu\",function(e){\n", + " return false;\n", + " });\n", + "\n", + " function set_focus () {\n", + " canvas.focus();\n", + " canvas_div.focus();\n", + " }\n", + "\n", + " window.setTimeout(set_focus, 100);\n", + "}\n", + "\n", + "mpl.figure.prototype._init_toolbar = function() {\n", + " var fig = this;\n", + "\n", + " var nav_element = $('
')\n", + " nav_element.attr('style', 'width: 100%');\n", + " this.root.append(nav_element);\n", + "\n", + " // Define a callback function for later on.\n", + " function toolbar_event(event) {\n", + " return fig.toolbar_button_onclick(event['data']);\n", + " }\n", + " function toolbar_mouse_event(event) {\n", + " return fig.toolbar_button_onmouseover(event['data']);\n", + " }\n", + "\n", + " for(var toolbar_ind in mpl.toolbar_items) {\n", + " var name = mpl.toolbar_items[toolbar_ind][0];\n", + " var tooltip = mpl.toolbar_items[toolbar_ind][1];\n", + " var image = mpl.toolbar_items[toolbar_ind][2];\n", + " var method_name = mpl.toolbar_items[toolbar_ind][3];\n", + "\n", + " if (!name) {\n", + " // put a spacer in here.\n", + " continue;\n", + " }\n", + " var button = $('