UBI_Notebook.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "    <div class=\"bk-root\">\n",
       "        <a href=\"https://bokeh.pydata.org\" target=\"_blank\" class=\"bk-logo bk-logo-small bk-logo-notebook\"></a>\n",
       "        <span id=\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\">Loading BokehJS ...</span>\n",
       "    </div>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/javascript": [
       "\n",
       "(function(root) {\n",
       "  function now() {\n",
       "    return new Date();\n",
       "  }\n",
       "\n",
       "  var force = true;\n",
       "\n",
       "  if (typeof (root._bokeh_onload_callbacks) === \"undefined\" || force === true) {\n",
       "    root._bokeh_onload_callbacks = [];\n",
       "    root._bokeh_is_loading = undefined;\n",
       "  }\n",
       "\n",
       "  var JS_MIME_TYPE = 'application/javascript';\n",
       "  var HTML_MIME_TYPE = 'text/html';\n",
       "  var EXEC_MIME_TYPE = 'application/vnd.bokehjs_exec.v0+json';\n",
       "  var CLASS_NAME = 'output_bokeh rendered_html';\n",
       "\n",
       "  /**\n",
       "   * Render data to the DOM node\n",
       "   */\n",
       "  function render(props, node) {\n",
       "    var script = document.createElement(\"script\");\n",
       "    node.appendChild(script);\n",
       "  }\n",
       "\n",
       "  /**\n",
       "   * Handle when an output is cleared or removed\n",
       "   */\n",
       "  function handleClearOutput(event, handle) {\n",
       "    var cell = handle.cell;\n",
       "\n",
       "    var id = cell.output_area._bokeh_element_id;\n",
       "    var server_id = cell.output_area._bokeh_server_id;\n",
       "    // Clean up Bokeh references\n",
       "    if (id !== undefined) {\n",
       "      Bokeh.index[id].model.document.clear();\n",
       "      delete Bokeh.index[id];\n",
       "    }\n",
       "\n",
       "    if (server_id !== undefined) {\n",
       "      // Clean up Bokeh references\n",
       "      var cmd = \"from bokeh.io.state import curstate; print(curstate().uuid_to_server['\" + server_id + \"'].get_sessions()[0].document.roots[0]._id)\";\n",
       "      cell.notebook.kernel.execute(cmd, {\n",
       "        iopub: {\n",
       "          output: function(msg) {\n",
       "            var element_id = msg.content.text.trim();\n",
       "            Bokeh.index[element_id].model.document.clear();\n",
       "            delete Bokeh.index[element_id];\n",
       "          }\n",
       "        }\n",
       "      });\n",
       "      // Destroy server and session\n",
       "      var cmd = \"import bokeh.io.notebook as ion; ion.destroy_server('\" + server_id + \"')\";\n",
       "      cell.notebook.kernel.execute(cmd);\n",
       "    }\n",
       "  }\n",
       "\n",
       "  /**\n",
       "   * Handle when a new output is added\n",
       "   */\n",
       "  function handleAddOutput(event, handle) {\n",
       "    var output_area = handle.output_area;\n",
       "    var output = handle.output;\n",
       "\n",
       "    // limit handleAddOutput to display_data with EXEC_MIME_TYPE content only\n",
       "    if ((output.output_type != \"display_data\") || (!output.data.hasOwnProperty(EXEC_MIME_TYPE))) {\n",
       "      return\n",
       "    }\n",
       "\n",
       "    var toinsert = output_area.element.find(\".\" + CLASS_NAME.split(' ')[0]);\n",
       "\n",
       "    if (output.metadata[EXEC_MIME_TYPE][\"id\"] !== undefined) {\n",
       "      toinsert[0].firstChild.textContent = output.data[JS_MIME_TYPE];\n",
       "      // store reference to embed id on output_area\n",
       "      output_area._bokeh_element_id = output.metadata[EXEC_MIME_TYPE][\"id\"];\n",
       "    }\n",
       "    if (output.metadata[EXEC_MIME_TYPE][\"server_id\"] !== undefined) {\n",
       "      var bk_div = document.createElement(\"div\");\n",
       "      bk_div.innerHTML = output.data[HTML_MIME_TYPE];\n",
       "      var script_attrs = bk_div.children[0].attributes;\n",
       "      for (var i = 0; i < script_attrs.length; i++) {\n",
       "        toinsert[0].firstChild.setAttribute(script_attrs[i].name, script_attrs[i].value);\n",
       "      }\n",
       "      // store reference to server id on output_area\n",
       "      output_area._bokeh_server_id = output.metadata[EXEC_MIME_TYPE][\"server_id\"];\n",
       "    }\n",
       "  }\n",
       "\n",
       "  function register_renderer(events, OutputArea) {\n",
       "\n",
       "    function append_mime(data, metadata, element) {\n",
       "      // create a DOM node to render to\n",
       "      var toinsert = this.create_output_subarea(\n",
       "        metadata,\n",
       "        CLASS_NAME,\n",
       "        EXEC_MIME_TYPE\n",
       "      );\n",
       "      this.keyboard_manager.register_events(toinsert);\n",
       "      // Render to node\n",
       "      var props = {data: data, metadata: metadata[EXEC_MIME_TYPE]};\n",
       "      render(props, toinsert[0]);\n",
       "      element.append(toinsert);\n",
       "      return toinsert\n",
       "    }\n",
       "\n",
       "    /* Handle when an output is cleared or removed */\n",
       "    events.on('clear_output.CodeCell', handleClearOutput);\n",
       "    events.on('delete.Cell', handleClearOutput);\n",
       "\n",
       "    /* Handle when a new output is added */\n",
       "    events.on('output_added.OutputArea', handleAddOutput);\n",
       "\n",
       "    /**\n",
       "     * Register the mime type and append_mime function with output_area\n",
       "     */\n",
       "    OutputArea.prototype.register_mime_type(EXEC_MIME_TYPE, append_mime, {\n",
       "      /* Is output safe? */\n",
       "      safe: true,\n",
       "      /* Index of renderer in `output_area.display_order` */\n",
       "      index: 0\n",
       "    });\n",
       "  }\n",
       "\n",
       "  // register the mime type if in Jupyter Notebook environment and previously unregistered\n",
       "  if (root.Jupyter !== undefined) {\n",
       "    var events = require('base/js/events');\n",
       "    var OutputArea = require('notebook/js/outputarea').OutputArea;\n",
       "\n",
       "    if (OutputArea.prototype.mime_types().indexOf(EXEC_MIME_TYPE) == -1) {\n",
       "      register_renderer(events, OutputArea);\n",
       "    }\n",
       "  }\n",
       "\n",
       "  \n",
       "  if (typeof (root._bokeh_timeout) === \"undefined\" || force === true) {\n",
       "    root._bokeh_timeout = Date.now() + 5000;\n",
       "    root._bokeh_failed_load = false;\n",
       "  }\n",
       "\n",
       "  var NB_LOAD_WARNING = {'data': {'text/html':\n",
       "     \"<div style='background-color: #fdd'>\\n\"+\n",
       "     \"<p>\\n\"+\n",
       "     \"BokehJS does not appear to have successfully loaded. If loading BokehJS from CDN, this \\n\"+\n",
       "     \"may be due to a slow or bad network connection. Possible fixes:\\n\"+\n",
       "     \"</p>\\n\"+\n",
       "     \"<ul>\\n\"+\n",
       "     \"<li>re-rerun `output_notebook()` to attempt to load from CDN again, or</li>\\n\"+\n",
       "     \"<li>use INLINE resources instead, as so:</li>\\n\"+\n",
       "     \"</ul>\\n\"+\n",
       "     \"<code>\\n\"+\n",
       "     \"from bokeh.resources import INLINE\\n\"+\n",
       "     \"output_notebook(resources=INLINE)\\n\"+\n",
       "     \"</code>\\n\"+\n",
       "     \"</div>\"}};\n",
       "\n",
       "  function display_loaded() {\n",
       "    var el = document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\");\n",
       "    if (el != null) {\n",
       "      el.textContent = \"BokehJS is loading...\";\n",
       "    }\n",
       "    if (root.Bokeh !== undefined) {\n",
       "      if (el != null) {\n",
       "        el.textContent = \"BokehJS \" + root.Bokeh.version + \" successfully loaded.\";\n",
       "      }\n",
       "    } else if (Date.now() < root._bokeh_timeout) {\n",
       "      setTimeout(display_loaded, 100)\n",
       "    }\n",
       "  }\n",
       "\n",
       "\n",
       "  function run_callbacks() {\n",
       "    try {\n",
       "      root._bokeh_onload_callbacks.forEach(function(callback) { callback() });\n",
       "    }\n",
       "    finally {\n",
       "      delete root._bokeh_onload_callbacks\n",
       "    }\n",
       "    console.info(\"Bokeh: all callbacks have finished\");\n",
       "  }\n",
       "\n",
       "  function load_libs(js_urls, callback) {\n",
       "    root._bokeh_onload_callbacks.push(callback);\n",
       "    if (root._bokeh_is_loading > 0) {\n",
       "      console.log(\"Bokeh: BokehJS is being loaded, scheduling callback at\", now());\n",
       "      return null;\n",
       "    }\n",
       "    if (js_urls == null || js_urls.length === 0) {\n",
       "      run_callbacks();\n",
       "      return null;\n",
       "    }\n",
       "    console.log(\"Bokeh: BokehJS not loaded, scheduling load and callback at\", now());\n",
       "    root._bokeh_is_loading = js_urls.length;\n",
       "    for (var i = 0; i < js_urls.length; i++) {\n",
       "      var url = js_urls[i];\n",
       "      var s = document.createElement('script');\n",
       "      s.src = url;\n",
       "      s.async = false;\n",
       "      s.onreadystatechange = s.onload = function() {\n",
       "        root._bokeh_is_loading--;\n",
       "        if (root._bokeh_is_loading === 0) {\n",
       "          console.log(\"Bokeh: all BokehJS libraries loaded\");\n",
       "          run_callbacks()\n",
       "        }\n",
       "      };\n",
       "      s.onerror = function() {\n",
       "        console.warn(\"failed to load library \" + url);\n",
       "      };\n",
       "      console.log(\"Bokeh: injecting script tag for BokehJS library: \", url);\n",
       "      document.getElementsByTagName(\"head\")[0].appendChild(s);\n",
       "    }\n",
       "  };var element = document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\");\n",
       "  if (element == null) {\n",
       "    console.log(\"Bokeh: ERROR: autoload.js configured with elementid '0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6' but no matching script tag was found. \")\n",
       "    return false;\n",
       "  }\n",
       "\n",
       "  var js_urls = [\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-gl-0.12.14.min.js\"];\n",
       "\n",
       "  var inline_js = [\n",
       "    function(Bokeh) {\n",
       "      Bokeh.set_log_level(\"info\");\n",
       "    },\n",
       "    \n",
       "    function(Bokeh) {\n",
       "      \n",
       "    },\n",
       "    function(Bokeh) {\n",
       "      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.css\");\n",
       "      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.css\");\n",
       "      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.css\");\n",
       "      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.css\");\n",
       "      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.css\");\n",
       "      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.css\");\n",
       "    }\n",
       "  ];\n",
       "\n",
       "  function run_inline_js() {\n",
       "    \n",
       "    if ((root.Bokeh !== undefined) || (force === true)) {\n",
       "      for (var i = 0; i < inline_js.length; i++) {\n",
       "        inline_js[i].call(root, root.Bokeh);\n",
       "      }if (force === true) {\n",
       "        display_loaded();\n",
       "      }} else if (Date.now() < root._bokeh_timeout) {\n",
       "      setTimeout(run_inline_js, 100);\n",
       "    } else if (!root._bokeh_failed_load) {\n",
       "      console.log(\"Bokeh: BokehJS failed to load within specified timeout.\");\n",
       "      root._bokeh_failed_load = true;\n",
       "    } else if (force !== true) {\n",
       "      var cell = $(document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\")).parents('.cell').data().cell;\n",
       "      cell.output_area.append_execute_result(NB_LOAD_WARNING)\n",
       "    }\n",
       "\n",
       "  }\n",
       "\n",
       "  if (root._bokeh_is_loading === 0) {\n",
       "    console.log(\"Bokeh: BokehJS loaded, going straight to plotting\");\n",
       "    run_inline_js();\n",
       "  } else {\n",
       "    load_libs(js_urls, function() {\n",
       "      console.log(\"Bokeh: BokehJS plotting callback run at\", now());\n",
       "      run_inline_js();\n",
       "    });\n",
       "  }\n",
       "}(window));"
      ],
      "application/vnd.bokehjs_load.v0+json": "\n(function(root) {\n  function now() {\n    return new Date();\n  }\n\n  var force = true;\n\n  if (typeof (root._bokeh_onload_callbacks) === \"undefined\" || force === true) {\n    root._bokeh_onload_callbacks = [];\n    root._bokeh_is_loading = undefined;\n  }\n\n  \n\n  \n  if (typeof (root._bokeh_timeout) === \"undefined\" || force === true) {\n    root._bokeh_timeout = Date.now() + 5000;\n    root._bokeh_failed_load = false;\n  }\n\n  var NB_LOAD_WARNING = {'data': {'text/html':\n     \"<div style='background-color: #fdd'>\\n\"+\n     \"<p>\\n\"+\n     \"BokehJS does not appear to have successfully loaded. If loading BokehJS from CDN, this \\n\"+\n     \"may be due to a slow or bad network connection. Possible fixes:\\n\"+\n     \"</p>\\n\"+\n     \"<ul>\\n\"+\n     \"<li>re-rerun `output_notebook()` to attempt to load from CDN again, or</li>\\n\"+\n     \"<li>use INLINE resources instead, as so:</li>\\n\"+\n     \"</ul>\\n\"+\n     \"<code>\\n\"+\n     \"from bokeh.resources import INLINE\\n\"+\n     \"output_notebook(resources=INLINE)\\n\"+\n     \"</code>\\n\"+\n     \"</div>\"}};\n\n  function display_loaded() {\n    var el = document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\");\n    if (el != null) {\n      el.textContent = \"BokehJS is loading...\";\n    }\n    if (root.Bokeh !== undefined) {\n      if (el != null) {\n        el.textContent = \"BokehJS \" + root.Bokeh.version + \" successfully loaded.\";\n      }\n    } else if (Date.now() < root._bokeh_timeout) {\n      setTimeout(display_loaded, 100)\n    }\n  }\n\n\n  function run_callbacks() {\n    try {\n      root._bokeh_onload_callbacks.forEach(function(callback) { callback() });\n    }\n    finally {\n      delete root._bokeh_onload_callbacks\n    }\n    console.info(\"Bokeh: all callbacks have finished\");\n  }\n\n  function load_libs(js_urls, callback) {\n    root._bokeh_onload_callbacks.push(callback);\n    if (root._bokeh_is_loading > 0) {\n      console.log(\"Bokeh: BokehJS is being loaded, scheduling callback at\", now());\n      return null;\n    }\n    if (js_urls == null || js_urls.length === 0) {\n      run_callbacks();\n      return null;\n    }\n    console.log(\"Bokeh: BokehJS not loaded, scheduling load and callback at\", now());\n    root._bokeh_is_loading = js_urls.length;\n    for (var i = 0; i < js_urls.length; i++) {\n      var url = js_urls[i];\n      var s = document.createElement('script');\n      s.src = url;\n      s.async = false;\n      s.onreadystatechange = s.onload = function() {\n        root._bokeh_is_loading--;\n        if (root._bokeh_is_loading === 0) {\n          console.log(\"Bokeh: all BokehJS libraries loaded\");\n          run_callbacks()\n        }\n      };\n      s.onerror = function() {\n        console.warn(\"failed to load library \" + url);\n      };\n      console.log(\"Bokeh: injecting script tag for BokehJS library: \", url);\n      document.getElementsByTagName(\"head\")[0].appendChild(s);\n    }\n  };var element = document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\");\n  if (element == null) {\n    console.log(\"Bokeh: ERROR: autoload.js configured with elementid '0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6' but no matching script tag was found. \")\n    return false;\n  }\n\n  var js_urls = [\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-gl-0.12.14.min.js\"];\n\n  var inline_js = [\n    function(Bokeh) {\n      Bokeh.set_log_level(\"info\");\n    },\n    \n    function(Bokeh) {\n      \n    },\n    function(Bokeh) {\n      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.css\");\n      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.14.min.css\");\n      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.css\");\n      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.14.min.css\");\n      console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.css\");\n      Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-tables-0.12.14.min.css\");\n    }\n  ];\n\n  function run_inline_js() {\n    \n    if ((root.Bokeh !== undefined) || (force === true)) {\n      for (var i = 0; i < inline_js.length; i++) {\n        inline_js[i].call(root, root.Bokeh);\n      }if (force === true) {\n        display_loaded();\n      }} else if (Date.now() < root._bokeh_timeout) {\n      setTimeout(run_inline_js, 100);\n    } else if (!root._bokeh_failed_load) {\n      console.log(\"Bokeh: BokehJS failed to load within specified timeout.\");\n      root._bokeh_failed_load = true;\n    } else if (force !== true) {\n      var cell = $(document.getElementById(\"0eb5546d-58ff-4c52-92c9-ce87cf8f1fc6\")).parents('.cell').data().cell;\n      cell.output_area.append_execute_result(NB_LOAD_WARNING)\n    }\n\n  }\n\n  if (root._bokeh_is_loading === 0) {\n    console.log(\"Bokeh: BokehJS loaded, going straight to plotting\");\n    run_inline_js();\n  } else {\n    load_libs(js_urls, function() {\n      console.log(\"Bokeh: BokehJS plotting callback run at\", now());\n      run_inline_js();\n    });\n  }\n}(window));"
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import taxcalc as tc\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import helpers\n",
    "import copy\n",
    "from bokeh.plotting import figure\n",
    "from bokeh.models import ColumnDataSource, HoverTool\n",
    "from bokeh.io import show, output_notebook\n",
    "output_notebook()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "start_yr = 2018\n",
    "end_yr = 2027\n",
    "years = [year for year in range(start_yr, end_yr + 1)]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Baseline Calculator"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "You loaded data for 2014.\n",
      "Tax-Calculator startup automatically extrapolated your data to 2014.\n"
     ]
    }
   ],
   "source": [
    "rec_base = tc.Records.cps_constructor()\n",
    "pol_base = tc.Policy()\n",
    "calc_base = tc.Calculator(records=rec_base, policy=pol_base)\n",
    "calc_base.advance_to_year(start_yr)\n",
    "calc_base.calc_all()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Tax and Benefit Reform"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "You loaded data for 2014.\n",
      "Tax-Calculator startup automatically extrapolated your data to 2014.\n"
     ]
    }
   ],
   "source": [
    "rec_reform = tc.Records.cps_constructor()\n",
    "pol_reform = tc.Policy()\n",
    "# read in and implement reform parameters\n",
    "params = tc.Calculator.read_json_param_objects('tax_reform.json',\n",
    "                                               assump=None)\n",
    "pol_reform.implement_reform(params['policy'])\n",
    "calc_reform = tc.Calculator(records=rec_reform, policy=pol_reform)\n",
    "calc_reform.advance_to_year(start_yr)\n",
    "calc_reform.calc_all()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Revenue Raised"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "tax_rev = calc_reform.weighted_total('combined') - calc_base.weighted_total('combined')\n",
    "ben_rev = calc_base.weighted_total('benefit_cost_total') - calc_reform.weighted_total('benefit_cost_total')\n",
    "total_rev = tax_rev + ben_rev"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Revenue Raised for 2018\n",
      "Total Tax Revenue Raised: $488649081187.16\n",
      "Total Benefit Revenue Raised: $2740908107827.55\n",
      "----------------------------------------------\n",
      "Total Revenue Raised: $3229557189014.7\n"
     ]
    }
   ],
   "source": [
    "print('Revenue Raised for 2018')\n",
    "print(f'Total Tax Revenue Raised: ${round(tax_rev, 2)}')\n",
    "print(f'Total Benefit Revenue Raised: ${round(ben_rev, 2)}')\n",
    "print('-' * 46)\n",
    "print(f'Total Revenue Raised: ${round(total_rev, 2)}')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Initial UBI\n",
    "\n",
    "Caculations for a UBI in 2018"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Base UBI for those under 18: $5499.9\n",
      "Base UBI for those above 18: $10999.8\n",
      "Note: these do not account for additional tax revenue due to the UBI\n"
     ]
    }
   ],
   "source": [
    "u18 = calc_reform.weighted_total('nu18')\n",
    "abv18 = calc_reform.weighted_total('n1820') + calc_reform.weighted_total('n21')\n",
    "ubi_18, ubi_u18 = helpers.ubi_amount(total_rev, u18, abv18)\n",
    "print(f'Base UBI for those under 18: ${round(ubi_u18, 2)}')\n",
    "print(f'Base UBI for those above 18: ${round(ubi_18, 1)}')\n",
    "print('Note: these do not account for additional tax revenue due to the UBI')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "You loaded data for 2014.\n",
      "Tax-Calculator startup automatically extrapolated your data to 2014.\n"
     ]
    }
   ],
   "source": [
    "# calculator to find initial UBI tax revenue\n",
    "rec_init = tc.Records.cps_constructor()\n",
    "pol_init = tc.Policy()\n",
    "pol_init.implement_reform(params['policy'])\n",
    "ubi_reform = {\n",
    "    2018: {\n",
    "        '_UBI_u18': [ubi_u18],\n",
    "        '_UBI_1820': [ubi_18],\n",
    "        '_UBI_21': [ubi_18]\n",
    "    }\n",
    "}\n",
    "pol_init.implement_reform(ubi_reform)\n",
    "calc_init = tc.Calculator(records=rec_init, policy=pol_init)\n",
    "calc_init.advance_to_year(2018)\n",
    "calc_init.calc_all()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Initial tax revenue raised by a UBI: $528323923334.34\n"
     ]
    }
   ],
   "source": [
    "reform_rev = calc_reform.weighted_total('combined')\n",
    "ubi_tax_rev_init = (calc_init.weighted_total('combined') -\n",
    "                   reform_rev)\n",
    "print(f'Initial tax revenue raised by a UBI: ${round(ubi_tax_rev_init, 2)}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "2018\n",
      "Iterations: 15\n",
      "Revenue Difference: $-40.07\n"
     ]
    }
   ],
   "source": [
    "# find UBI after accounting for additional tax revenue\n",
    "ubi_18_final, ubi_u18_final = helpers.ubi_finder(u18, abv18, ubi_18, ubi_u18,\n",
    "                                                 params, total_rev, reform_rev,\n",
    "                                                 2018, 100)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Budget Neutral UBI Amounts:\n",
      "\t Those under 18: $6606.85\n",
      "\t Those above 18: $13213.7\n"
     ]
    }
   ],
   "source": [
    "print('Budget Neutral UBI Amounts:')\n",
    "print(f'\\t Those under 18: ${round(ubi_u18_final, 2)}')\n",
    "print(f'\\t Those above 18: ${round(ubi_18_final, 2)}')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Calculations with a UBI"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "You loaded data for 2014.\n",
      "Tax-Calculator startup automatically extrapolated your data to 2014.\n"
     ]
    }
   ],
   "source": [
    "rec_ubi = tc.Records.cps_constructor()\n",
    "pol_ubi = tc.Policy()\n",
    "pol_ubi.implement_reform(params['policy'])\n",
    "ubi_reform = {\n",
    "    2018: {\n",
    "        '_UBI_u18': [ubi_u18_final],\n",
    "        '_UBI_1820': [ubi_18_final],\n",
    "        '_UBI_21': [ubi_18_final]\n",
    "    }\n",
    "}\n",
    "pol_ubi.implement_reform(ubi_reform)\n",
    "calc_ubi = tc.Calculator(records=rec_ubi, policy = pol_ubi)\n",
    "calc_ubi.advance_to_year(2018)\n",
    "calc_ubi.calc_all()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "# UBI table data prep\n",
    "table_data = calc_ubi.dataframe(['ubi', 'e00200', 'nu18', 'n1820', 'n21',\n",
    "                                 'combined', 's006', 'benefit_cost_total',\n",
    "                                 'benefit_value_total', 'elderly_dependent',\n",
    "                                 'age_head', 'age_spouse'])\n",
    "mtrs = calc_ubi.mtr()\n",
    "table_data['pmtr'] = mtrs[0]\n",
    "table_data['imtr'] = mtrs[1]\n",
    "table_data['cmtr'] = mtrs[2]\n",
    "table_data['ppl'] = table_data[['nu18', 'n1820', 'n21']].sum(axis=0)\n",
    "table_data['ubi/person'] = table_data['ubi'] / table_data['ppl']\n",
    "tax_change = calc_ubi.array('combined') - calc_base.array('combined')\n",
    "table_data['tax_change'] = tax_change\n",
    "ben_change = (calc_ubi.array('benefit_cost_total') -\n",
    "              calc_base.array('benefit_cost_total'))\n",
    "table_data['ben_change'] = ben_change\n",
    "ben_change_welfare = (calc_ubi.array('benefit_value_total') -\n",
    "                      calc_base.array('benefit_value_total'))\n",
    "table_data['ben_change_welfare'] = ben_change_welfare\n",
    "combined_change = (table_data['ubi'] -\n",
    "                   table_data['tax_change'] +\n",
    "                   table_data['ben_change'])\n",
    "table_data['combined_change'] = combined_change\n",
    "combined_change_welfare = (table_data['ubi'] -\n",
    "                           table_data['tax_change'] +\n",
    "                           table_data['ben_change_welfare'])\n",
    "table_data['combined_change_welfare'] = combined_change_welfare\n",
    "table_data['65_plus'] = np.where((table_data['elderly_dependent'] == 1) |\n",
    "                                  (table_data['age_head'] >= 65) |\n",
    "                                  (table_data['age_spouse'] >= 65), 1, 0)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# All Tax Units"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "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></th>\n",
       "      <th>Wage and Salary Floor (Thousands)</th>\n",
       "      <th>Tax Units (m)</th>\n",
       "      <th>Avg Tax Unit Size</th>\n",
       "      <th>Avg UBI Per Person</th>\n",
       "      <th>Avg MTR - Payroll</th>\n",
       "      <th>Avg MTR - Individual Income</th>\n",
       "      <th>Avg MTR - Combined</th>\n",
       "      <th>Avg Tax Change</th>\n",
       "      <th>Avg UBI Per Tax Unit</th>\n",
       "      <th>Avg Benefits Change</th>\n",
       "      <th>Avg Benefits Change - Welfare Adj</th>\n",
       "      <th>Avg Combined Change</th>\n",
       "      <th>Avg Combined Change - Welfare Adj</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>45.954719</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>13.3%</td>\n",
       "      <td>27.5%</td>\n",
       "      <td>2,974</td>\n",
       "      <td>19,676</td>\n",
       "      <td>-39,286</td>\n",
       "      <td>-39,286</td>\n",
       "      <td>-22,584</td>\n",
       "      <td>-22,584</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>15.914526</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>10.3%</td>\n",
       "      <td>24.5%</td>\n",
       "      <td>2,974</td>\n",
       "      <td>16,907</td>\n",
       "      <td>-11,644</td>\n",
       "      <td>-11,644</td>\n",
       "      <td>2,289</td>\n",
       "      <td>2,289</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>10</td>\n",
       "      <td>14.414201</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>12.1%</td>\n",
       "      <td>26.3%</td>\n",
       "      <td>5,543</td>\n",
       "      <td>20,383</td>\n",
       "      <td>-11,620</td>\n",
       "      <td>-11,620</td>\n",
       "      <td>3,220</td>\n",
       "      <td>3,220</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>20</td>\n",
       "      <td>15.253706</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>14.8%</td>\n",
       "      <td>29.0%</td>\n",
       "      <td>6,201</td>\n",
       "      <td>21,381</td>\n",
       "      <td>-9,220</td>\n",
       "      <td>-9,220</td>\n",
       "      <td>5,960</td>\n",
       "      <td>5,960</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>30</td>\n",
       "      <td>12.411297</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>18.2%</td>\n",
       "      <td>32.4%</td>\n",
       "      <td>6,476</td>\n",
       "      <td>22,027</td>\n",
       "      <td>-7,584</td>\n",
       "      <td>-7,584</td>\n",
       "      <td>7,967</td>\n",
       "      <td>7,967</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>40</td>\n",
       "      <td>10.622800</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>18.8%</td>\n",
       "      <td>33.0%</td>\n",
       "      <td>6,999</td>\n",
       "      <td>23,218</td>\n",
       "      <td>-7,133</td>\n",
       "      <td>-7,133</td>\n",
       "      <td>9,086</td>\n",
       "      <td>9,086</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>50</td>\n",
       "      <td>18.995208</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>20.4%</td>\n",
       "      <td>34.6%</td>\n",
       "      <td>8,111</td>\n",
       "      <td>24,941</td>\n",
       "      <td>-6,129</td>\n",
       "      <td>-6,129</td>\n",
       "      <td>10,701</td>\n",
       "      <td>10,701</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>75</td>\n",
       "      <td>12.312157</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>21.1%</td>\n",
       "      <td>35.3%</td>\n",
       "      <td>10,448</td>\n",
       "      <td>27,773</td>\n",
       "      <td>-4,785</td>\n",
       "      <td>-4,785</td>\n",
       "      <td>12,541</td>\n",
       "      <td>12,541</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>100</td>\n",
       "      <td>18.660672</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>12.6%</td>\n",
       "      <td>22.3%</td>\n",
       "      <td>34.9%</td>\n",
       "      <td>13,239</td>\n",
       "      <td>30,858</td>\n",
       "      <td>-4,099</td>\n",
       "      <td>-4,099</td>\n",
       "      <td>13,520</td>\n",
       "      <td>13,520</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>200</td>\n",
       "      <td>4.954667</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>7.9%</td>\n",
       "      <td>26.9%</td>\n",
       "      <td>34.8%</td>\n",
       "      <td>17,099</td>\n",
       "      <td>32,668</td>\n",
       "      <td>-3,755</td>\n",
       "      <td>-3,755</td>\n",
       "      <td>11,813</td>\n",
       "      <td>11,813</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>500</td>\n",
       "      <td>0.289244</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>6.5%</td>\n",
       "      <td>35.6%</td>\n",
       "      <td>42.1%</td>\n",
       "      <td>25,575</td>\n",
       "      <td>32,404</td>\n",
       "      <td>-4,373</td>\n",
       "      <td>-4,373</td>\n",
       "      <td>2,456</td>\n",
       "      <td>2,456</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>1,000</td>\n",
       "      <td>0.111645</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>6.3%</td>\n",
       "      <td>36.0%</td>\n",
       "      <td>42.3%</td>\n",
       "      <td>31,837</td>\n",
       "      <td>31,776</td>\n",
       "      <td>-4,404</td>\n",
       "      <td>-4,404</td>\n",
       "      <td>-4,465</td>\n",
       "      <td>-4,465</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Wage and Salary Floor (Thousands)  Tax Units (m)  Avg Tax Unit Size  \\\n",
       "0                                  0      45.954719                0.0   \n",
       "1                                  0      15.914526                0.0   \n",
       "2                                 10      14.414201                0.0   \n",
       "3                                 20      15.253706                0.0   \n",
       "4                                 30      12.411297                0.0   \n",
       "5                                 40      10.622800                0.0   \n",
       "6                                 50      18.995208                0.0   \n",
       "7                                 75      12.312157                0.0   \n",
       "8                                100      18.660672                0.0   \n",
       "9                                200       4.954667                0.0   \n",
       "10                               500       0.289244                0.0   \n",
       "11                             1,000       0.111645                0.0   \n",
       "\n",
       "   Avg UBI Per Person Avg MTR - Payroll Avg MTR - Individual Income  \\\n",
       "0                   0             14.2%                       13.3%   \n",
       "1                   0             14.2%                       10.3%   \n",
       "2                   0             14.2%                       12.1%   \n",
       "3                   0             14.2%                       14.8%   \n",
       "4                   0             14.2%                       18.2%   \n",
       "5                   0             14.2%                       18.8%   \n",
       "6                   0             14.2%                       20.4%   \n",
       "7                   0             14.2%                       21.1%   \n",
       "8                   0             12.6%                       22.3%   \n",
       "9                   0              7.9%                       26.9%   \n",
       "10                  0              6.5%                       35.6%   \n",
       "11                  0              6.3%                       36.0%   \n",
       "\n",
       "   Avg MTR - Combined Avg Tax Change Avg UBI Per Tax Unit Avg Benefits Change  \\\n",
       "0               27.5%          2,974               19,676             -39,286   \n",
       "1               24.5%          2,974               16,907             -11,644   \n",
       "2               26.3%          5,543               20,383             -11,620   \n",
       "3               29.0%          6,201               21,381              -9,220   \n",
       "4               32.4%          6,476               22,027              -7,584   \n",
       "5               33.0%          6,999               23,218              -7,133   \n",
       "6               34.6%          8,111               24,941              -6,129   \n",
       "7               35.3%         10,448               27,773              -4,785   \n",
       "8               34.9%         13,239               30,858              -4,099   \n",
       "9               34.8%         17,099               32,668              -3,755   \n",
       "10              42.1%         25,575               32,404              -4,373   \n",
       "11              42.3%         31,837               31,776              -4,404   \n",
       "\n",
       "   Avg Benefits Change - Welfare Adj Avg Combined Change  \\\n",
       "0                            -39,286             -22,584   \n",
       "1                            -11,644               2,289   \n",
       "2                            -11,620               3,220   \n",
       "3                             -9,220               5,960   \n",
       "4                             -7,584               7,967   \n",
       "5                             -7,133               9,086   \n",
       "6                             -6,129              10,701   \n",
       "7                             -4,785              12,541   \n",
       "8                             -4,099              13,520   \n",
       "9                             -3,755              11,813   \n",
       "10                            -4,373               2,456   \n",
       "11                            -4,404              -4,465   \n",
       "\n",
       "   Avg Combined Change - Welfare Adj  \n",
       "0                            -22,584  \n",
       "1                              2,289  \n",
       "2                              3,220  \n",
       "3                              5,960  \n",
       "4                              7,967  \n",
       "5                              9,086  \n",
       "6                             10,701  \n",
       "7                             12,541  \n",
       "8                             13,520  \n",
       "9                             11,813  \n",
       "10                             2,456  \n",
       "11                            -4,465  "
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "helpers.table(table_data, 'e00200', 'webapp')  # will need to change 'webapp' to 'standard'"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Tax Units with Someone 65 or Older"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "elderly_data = copy.deepcopy(table_data[table_data['65_plus'] == 1])"
   ]
  },
  {
   "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></th>\n",
       "      <th>Wage and Salary Floor (Thousands)</th>\n",
       "      <th>Tax Units (m)</th>\n",
       "      <th>Avg Tax Unit Size</th>\n",
       "      <th>Avg UBI Per Person</th>\n",
       "      <th>Avg MTR - Payroll</th>\n",
       "      <th>Avg MTR - Individual Income</th>\n",
       "      <th>Avg MTR - Combined</th>\n",
       "      <th>Avg Tax Change</th>\n",
       "      <th>Avg UBI Per Tax Unit</th>\n",
       "      <th>Avg Benefits Change</th>\n",
       "      <th>Avg Benefits Change - Welfare Adj</th>\n",
       "      <th>Avg Combined Change</th>\n",
       "      <th>Avg Combined Change - Welfare Adj</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>25.842219</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>14.4%</td>\n",
       "      <td>28.6%</td>\n",
       "      <td>2,559</td>\n",
       "      <td>19,347</td>\n",
       "      <td>-47,220</td>\n",
       "      <td>-47,220</td>\n",
       "      <td>-30,433</td>\n",
       "      <td>-30,433</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>1.599303</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>13.1%</td>\n",
       "      <td>27.3%</td>\n",
       "      <td>3,628</td>\n",
       "      <td>21,524</td>\n",
       "      <td>-43,137</td>\n",
       "      <td>-43,137</td>\n",
       "      <td>-25,241</td>\n",
       "      <td>-25,241</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>10</td>\n",
       "      <td>1.413394</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>13.1%</td>\n",
       "      <td>27.3%</td>\n",
       "      <td>4,623</td>\n",
       "      <td>22,429</td>\n",
       "      <td>-40,500</td>\n",
       "      <td>-40,500</td>\n",
       "      <td>-22,694</td>\n",
       "      <td>-22,694</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>20</td>\n",
       "      <td>1.208057</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>14.6%</td>\n",
       "      <td>28.8%</td>\n",
       "      <td>5,097</td>\n",
       "      <td>23,322</td>\n",
       "      <td>-37,524</td>\n",
       "      <td>-37,524</td>\n",
       "      <td>-19,299</td>\n",
       "      <td>-19,299</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>30</td>\n",
       "      <td>0.931875</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>16.7%</td>\n",
       "      <td>30.9%</td>\n",
       "      <td>5,265</td>\n",
       "      <td>24,119</td>\n",
       "      <td>-33,561</td>\n",
       "      <td>-33,561</td>\n",
       "      <td>-14,707</td>\n",
       "      <td>-14,707</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>40</td>\n",
       "      <td>0.817449</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>17.8%</td>\n",
       "      <td>32.0%</td>\n",
       "      <td>5,316</td>\n",
       "      <td>24,656</td>\n",
       "      <td>-36,161</td>\n",
       "      <td>-36,161</td>\n",
       "      <td>-16,820</td>\n",
       "      <td>-16,820</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>50</td>\n",
       "      <td>1.552293</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>20.5%</td>\n",
       "      <td>34.7%</td>\n",
       "      <td>6,070</td>\n",
       "      <td>25,124</td>\n",
       "      <td>-33,428</td>\n",
       "      <td>-33,428</td>\n",
       "      <td>-14,374</td>\n",
       "      <td>-14,374</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>75</td>\n",
       "      <td>0.874385</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>21.1%</td>\n",
       "      <td>35.3%</td>\n",
       "      <td>7,804</td>\n",
       "      <td>26,398</td>\n",
       "      <td>-28,962</td>\n",
       "      <td>-28,962</td>\n",
       "      <td>-10,368</td>\n",
       "      <td>-10,368</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>100</td>\n",
       "      <td>1.333232</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>12.3%</td>\n",
       "      <td>22.7%</td>\n",
       "      <td>35.0%</td>\n",
       "      <td>9,097</td>\n",
       "      <td>27,483</td>\n",
       "      <td>-28,514</td>\n",
       "      <td>-28,514</td>\n",
       "      <td>-10,128</td>\n",
       "      <td>-10,128</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>200</td>\n",
       "      <td>0.384326</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0%</td>\n",
       "      <td>27.6%</td>\n",
       "      <td>35.5%</td>\n",
       "      <td>11,877</td>\n",
       "      <td>28,342</td>\n",
       "      <td>-24,673</td>\n",
       "      <td>-24,673</td>\n",
       "      <td>-8,208</td>\n",
       "      <td>-8,208</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>500</td>\n",
       "      <td>0.026964</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>7.0%</td>\n",
       "      <td>35.5%</td>\n",
       "      <td>42.5%</td>\n",
       "      <td>17,138</td>\n",
       "      <td>27,580</td>\n",
       "      <td>-23,644</td>\n",
       "      <td>-23,644</td>\n",
       "      <td>-13,202</td>\n",
       "      <td>-13,202</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>1,001</td>\n",
       "      <td>0.009955</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>6.9%</td>\n",
       "      <td>35.9%</td>\n",
       "      <td>42.7%</td>\n",
       "      <td>22,140</td>\n",
       "      <td>26,465</td>\n",
       "      <td>-22,625</td>\n",
       "      <td>-22,625</td>\n",
       "      <td>-18,300</td>\n",
       "      <td>-18,300</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Wage and Salary Floor (Thousands)  Tax Units (m)  Avg Tax Unit Size  \\\n",
       "0                                  0      25.842219                0.0   \n",
       "1                                  0       1.599303                0.0   \n",
       "2                                 10       1.413394                0.0   \n",
       "3                                 20       1.208057                0.0   \n",
       "4                                 30       0.931875                0.0   \n",
       "5                                 40       0.817449                0.0   \n",
       "6                                 50       1.552293                0.0   \n",
       "7                                 75       0.874385                0.0   \n",
       "8                                100       1.333232                0.0   \n",
       "9                                200       0.384326                0.0   \n",
       "10                               500       0.026964                0.0   \n",
       "11                             1,001       0.009955                0.0   \n",
       "\n",
       "   Avg UBI Per Person Avg MTR - Payroll Avg MTR - Individual Income  \\\n",
       "0                   0             14.2%                       14.4%   \n",
       "1                   0             14.2%                       13.1%   \n",
       "2                   0             14.2%                       13.1%   \n",
       "3                   0             14.2%                       14.6%   \n",
       "4                   0             14.2%                       16.7%   \n",
       "5                   0             14.2%                       17.8%   \n",
       "6                   0             14.2%                       20.5%   \n",
       "7                   0             14.2%                       21.1%   \n",
       "8                   0             12.3%                       22.7%   \n",
       "9                   0              8.0%                       27.6%   \n",
       "10                  0              7.0%                       35.5%   \n",
       "11                  0              6.9%                       35.9%   \n",
       "\n",
       "   Avg MTR - Combined Avg Tax Change Avg UBI Per Tax Unit Avg Benefits Change  \\\n",
       "0               28.6%          2,559               19,347             -47,220   \n",
       "1               27.3%          3,628               21,524             -43,137   \n",
       "2               27.3%          4,623               22,429             -40,500   \n",
       "3               28.8%          5,097               23,322             -37,524   \n",
       "4               30.9%          5,265               24,119             -33,561   \n",
       "5               32.0%          5,316               24,656             -36,161   \n",
       "6               34.7%          6,070               25,124             -33,428   \n",
       "7               35.3%          7,804               26,398             -28,962   \n",
       "8               35.0%          9,097               27,483             -28,514   \n",
       "9               35.5%         11,877               28,342             -24,673   \n",
       "10              42.5%         17,138               27,580             -23,644   \n",
       "11              42.7%         22,140               26,465             -22,625   \n",
       "\n",
       "   Avg Benefits Change - Welfare Adj Avg Combined Change  \\\n",
       "0                            -47,220             -30,433   \n",
       "1                            -43,137             -25,241   \n",
       "2                            -40,500             -22,694   \n",
       "3                            -37,524             -19,299   \n",
       "4                            -33,561             -14,707   \n",
       "5                            -36,161             -16,820   \n",
       "6                            -33,428             -14,374   \n",
       "7                            -28,962             -10,368   \n",
       "8                            -28,514             -10,128   \n",
       "9                            -24,673              -8,208   \n",
       "10                           -23,644             -13,202   \n",
       "11                           -22,625             -18,300   \n",
       "\n",
       "   Avg Combined Change - Welfare Adj  \n",
       "0                            -30,433  \n",
       "1                            -25,241  \n",
       "2                            -22,694  \n",
       "3                            -19,299  \n",
       "4                            -14,707  \n",
       "5                            -16,820  \n",
       "6                            -14,374  \n",
       "7                            -10,368  \n",
       "8                            -10,128  \n",
       "9                             -8,208  \n",
       "10                           -13,202  \n",
       "11                           -18,300  "
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "helpers.table(elderly_data, 'e00200', 'webapp')  # will need to change 'webapp' to 'standard'"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Tax Units without Someone 65 or Older"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "non_elderly_data = copy.deepcopy(table_data[table_data['65_plus'] == 0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "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></th>\n",
       "      <th>Wage and Salary Floor (Thousands)</th>\n",
       "      <th>Tax Units (m)</th>\n",
       "      <th>Avg Tax Unit Size</th>\n",
       "      <th>Avg UBI Per Person</th>\n",
       "      <th>Avg MTR - Payroll</th>\n",
       "      <th>Avg MTR - Individual Income</th>\n",
       "      <th>Avg MTR - Combined</th>\n",
       "      <th>Avg Tax Change</th>\n",
       "      <th>Avg UBI Per Tax Unit</th>\n",
       "      <th>Avg Benefits Change</th>\n",
       "      <th>Avg Benefits Change - Welfare Adj</th>\n",
       "      <th>Avg Combined Change</th>\n",
       "      <th>Avg Combined Change - Welfare Adj</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>20.112499</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.1%</td>\n",
       "      <td>11.9%</td>\n",
       "      <td>26.1%</td>\n",
       "      <td>3,508</td>\n",
       "      <td>20,099</td>\n",
       "      <td>-29,091</td>\n",
       "      <td>-29,091</td>\n",
       "      <td>-12,500</td>\n",
       "      <td>-12,500</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>14.315223</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>10.0%</td>\n",
       "      <td>24.2%</td>\n",
       "      <td>2,901</td>\n",
       "      <td>16,391</td>\n",
       "      <td>-8,125</td>\n",
       "      <td>-8,125</td>\n",
       "      <td>5,365</td>\n",
       "      <td>5,365</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>10</td>\n",
       "      <td>13.000807</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>12.0%</td>\n",
       "      <td>26.2%</td>\n",
       "      <td>5,643</td>\n",
       "      <td>20,161</td>\n",
       "      <td>-8,480</td>\n",
       "      <td>-8,480</td>\n",
       "      <td>6,037</td>\n",
       "      <td>6,037</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>20</td>\n",
       "      <td>14.045649</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>14.8%</td>\n",
       "      <td>29.0%</td>\n",
       "      <td>6,296</td>\n",
       "      <td>21,214</td>\n",
       "      <td>-6,786</td>\n",
       "      <td>-6,786</td>\n",
       "      <td>8,132</td>\n",
       "      <td>8,132</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>30</td>\n",
       "      <td>11.479422</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>18.3%</td>\n",
       "      <td>32.5%</td>\n",
       "      <td>6,575</td>\n",
       "      <td>21,858</td>\n",
       "      <td>-5,476</td>\n",
       "      <td>-5,476</td>\n",
       "      <td>9,807</td>\n",
       "      <td>9,807</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>40</td>\n",
       "      <td>9.805351</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>18.9%</td>\n",
       "      <td>33.1%</td>\n",
       "      <td>7,139</td>\n",
       "      <td>23,098</td>\n",
       "      <td>-4,713</td>\n",
       "      <td>-4,713</td>\n",
       "      <td>11,246</td>\n",
       "      <td>11,246</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>50</td>\n",
       "      <td>17.442915</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>20.4%</td>\n",
       "      <td>34.6%</td>\n",
       "      <td>8,292</td>\n",
       "      <td>24,925</td>\n",
       "      <td>-3,700</td>\n",
       "      <td>-3,700</td>\n",
       "      <td>12,932</td>\n",
       "      <td>12,932</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>75</td>\n",
       "      <td>11.437771</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>14.2%</td>\n",
       "      <td>21.2%</td>\n",
       "      <td>35.3%</td>\n",
       "      <td>10,650</td>\n",
       "      <td>27,878</td>\n",
       "      <td>-2,936</td>\n",
       "      <td>-2,936</td>\n",
       "      <td>14,292</td>\n",
       "      <td>14,292</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>100</td>\n",
       "      <td>17.327440</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>12.6%</td>\n",
       "      <td>22.3%</td>\n",
       "      <td>34.9%</td>\n",
       "      <td>13,558</td>\n",
       "      <td>31,118</td>\n",
       "      <td>-2,221</td>\n",
       "      <td>-2,221</td>\n",
       "      <td>15,339</td>\n",
       "      <td>15,339</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>200</td>\n",
       "      <td>4.570341</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>7.9%</td>\n",
       "      <td>26.8%</td>\n",
       "      <td>34.7%</td>\n",
       "      <td>17,539</td>\n",
       "      <td>33,031</td>\n",
       "      <td>-1,996</td>\n",
       "      <td>-1,996</td>\n",
       "      <td>13,497</td>\n",
       "      <td>13,497</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>500</td>\n",
       "      <td>0.262280</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>6.5%</td>\n",
       "      <td>35.6%</td>\n",
       "      <td>42.1%</td>\n",
       "      <td>26,442</td>\n",
       "      <td>32,900</td>\n",
       "      <td>-2,391</td>\n",
       "      <td>-2,391</td>\n",
       "      <td>4,066</td>\n",
       "      <td>4,066</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>1,000</td>\n",
       "      <td>0.101690</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>6.3%</td>\n",
       "      <td>36.0%</td>\n",
       "      <td>42.3%</td>\n",
       "      <td>32,787</td>\n",
       "      <td>32,296</td>\n",
       "      <td>-2,620</td>\n",
       "      <td>-2,620</td>\n",
       "      <td>-3,111</td>\n",
       "      <td>-3,111</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Wage and Salary Floor (Thousands)  Tax Units (m)  Avg Tax Unit Size  \\\n",
       "0                                  0      20.112499                0.0   \n",
       "1                                  0      14.315223                0.0   \n",
       "2                                 10      13.000807                0.0   \n",
       "3                                 20      14.045649                0.0   \n",
       "4                                 30      11.479422                0.0   \n",
       "5                                 40       9.805351                0.0   \n",
       "6                                 50      17.442915                0.0   \n",
       "7                                 75      11.437771                0.0   \n",
       "8                                100      17.327440                0.0   \n",
       "9                                200       4.570341                0.0   \n",
       "10                               500       0.262280                0.0   \n",
       "11                             1,000       0.101690                0.0   \n",
       "\n",
       "   Avg UBI Per Person Avg MTR - Payroll Avg MTR - Individual Income  \\\n",
       "0                   0             14.1%                       11.9%   \n",
       "1                   0             14.2%                       10.0%   \n",
       "2                   0             14.2%                       12.0%   \n",
       "3                   0             14.2%                       14.8%   \n",
       "4                   0             14.2%                       18.3%   \n",
       "5                   0             14.2%                       18.9%   \n",
       "6                   0             14.2%                       20.4%   \n",
       "7                   0             14.2%                       21.2%   \n",
       "8                   0             12.6%                       22.3%   \n",
       "9                   0              7.9%                       26.8%   \n",
       "10                  0              6.5%                       35.6%   \n",
       "11                  0              6.3%                       36.0%   \n",
       "\n",
       "   Avg MTR - Combined Avg Tax Change Avg UBI Per Tax Unit Avg Benefits Change  \\\n",
       "0               26.1%          3,508               20,099             -29,091   \n",
       "1               24.2%          2,901               16,391              -8,125   \n",
       "2               26.2%          5,643               20,161              -8,480   \n",
       "3               29.0%          6,296               21,214              -6,786   \n",
       "4               32.5%          6,575               21,858              -5,476   \n",
       "5               33.1%          7,139               23,098              -4,713   \n",
       "6               34.6%          8,292               24,925              -3,700   \n",
       "7               35.3%         10,650               27,878              -2,936   \n",
       "8               34.9%         13,558               31,118              -2,221   \n",
       "9               34.7%         17,539               33,031              -1,996   \n",
       "10              42.1%         26,442               32,900              -2,391   \n",
       "11              42.3%         32,787               32,296              -2,620   \n",
       "\n",
       "   Avg Benefits Change - Welfare Adj Avg Combined Change  \\\n",
       "0                            -29,091             -12,500   \n",
       "1                             -8,125               5,365   \n",
       "2                             -8,480               6,037   \n",
       "3                             -6,786               8,132   \n",
       "4                             -5,476               9,807   \n",
       "5                             -4,713              11,246   \n",
       "6                             -3,700              12,932   \n",
       "7                             -2,936              14,292   \n",
       "8                             -2,221              15,339   \n",
       "9                             -1,996              13,497   \n",
       "10                            -2,391               4,066   \n",
       "11                            -2,620              -3,111   \n",
       "\n",
       "   Avg Combined Change - Welfare Adj  \n",
       "0                            -12,500  \n",
       "1                              5,365  \n",
       "2                              6,037  \n",
       "3                              8,132  \n",
       "4                              9,807  \n",
       "5                             11,246  \n",
       "6                             12,932  \n",
       "7                             14,292  \n",
       "8                             15,339  \n",
       "9                             13,497  \n",
       "10                             4,066  \n",
       "11                            -3,111  "
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "helpers.table(non_elderly_data, 'e00200', 'webapp')  # will need to change 'webapp' to 'standard'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python [conda env:ubipaper]",
   "language": "python",
   "name": "conda-env-ubipaper-py"
  },
  "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.6.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}