v2_python.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import cvxpy as cp\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "from vizlib import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 125,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<style type=\"text/css\">\n",
       "#T_bae3e_row0_col1, #T_bae3e_row2_col1, #T_bae3e_row4_col1, #T_bae3e_row5_col1, #T_bae3e_row8_col1, #T_bae3e_row9_col1, #T_bae3e_row11_col1, #T_bae3e_row13_col1, #T_bae3e_row14_col1, #T_bae3e_row19_col1 {\n",
       "  background-color: peru;\n",
       "}\n",
       "#T_bae3e_row0_col2, #T_bae3e_row6_col2, #T_bae3e_row7_col2, #T_bae3e_row12_col2, #T_bae3e_row14_col2, #T_bae3e_row15_col2, #T_bae3e_row17_col2, #T_bae3e_row18_col2, #T_bae3e_row20_col2 {\n",
       "  background-color: red;\n",
       "}\n",
       "#T_bae3e_row0_col3, #T_bae3e_row0_col6, #T_bae3e_row1_col1, #T_bae3e_row1_col2, #T_bae3e_row1_col3, #T_bae3e_row1_col8, #T_bae3e_row2_col2, #T_bae3e_row2_col3, #T_bae3e_row2_col8, #T_bae3e_row3_col1, #T_bae3e_row3_col2, #T_bae3e_row3_col3, #T_bae3e_row4_col2, #T_bae3e_row4_col3, #T_bae3e_row4_col7, #T_bae3e_row5_col3, #T_bae3e_row5_col6, #T_bae3e_row5_col7, #T_bae3e_row6_col1, #T_bae3e_row6_col3, #T_bae3e_row6_col4, #T_bae3e_row7_col1, #T_bae3e_row7_col3, #T_bae3e_row7_col4, #T_bae3e_row7_col8, #T_bae3e_row8_col2, #T_bae3e_row8_col4, #T_bae3e_row8_col8, #T_bae3e_row9_col2, #T_bae3e_row9_col3, #T_bae3e_row9_col6, #T_bae3e_row10_col2, #T_bae3e_row10_col3, #T_bae3e_row10_col6, #T_bae3e_row11_col2, #T_bae3e_row11_col3, #T_bae3e_row11_col6, #T_bae3e_row12_col3, #T_bae3e_row12_col5, #T_bae3e_row12_col6, #T_bae3e_row13_col3, #T_bae3e_row13_col5, #T_bae3e_row14_col4, #T_bae3e_row15_col0, #T_bae3e_row15_col1, #T_bae3e_row15_col4, #T_bae3e_row16_col1, #T_bae3e_row16_col8, #T_bae3e_row17_col1, #T_bae3e_row17_col3, #T_bae3e_row17_col8, #T_bae3e_row18_col1, #T_bae3e_row18_col3, #T_bae3e_row19_col4, #T_bae3e_row19_col6, #T_bae3e_row20_col3, #T_bae3e_row20_col5, #T_bae3e_row20_col6 {\n",
       "  background-color: gray;\n",
       "}\n",
       "#T_bae3e_row0_col7, #T_bae3e_row2_col7, #T_bae3e_row3_col7, #T_bae3e_row9_col7, #T_bae3e_row10_col7, #T_bae3e_row12_col7, #T_bae3e_row14_col7, #T_bae3e_row16_col7, #T_bae3e_row17_col7, #T_bae3e_row20_col7 {\n",
       "  background-color: orange;\n",
       "}\n",
       "#T_bae3e_row0_col8, #T_bae3e_row3_col8, #T_bae3e_row5_col8, #T_bae3e_row6_col8, #T_bae3e_row9_col8, #T_bae3e_row11_col8, #T_bae3e_row13_col8, #T_bae3e_row15_col8, #T_bae3e_row19_col8, #T_bae3e_row20_col8 {\n",
       "  background-color: magenta;\n",
       "}\n",
       "#T_bae3e_row1_col0, #T_bae3e_row3_col0, #T_bae3e_row4_col0, #T_bae3e_row6_col0, #T_bae3e_row7_col0, #T_bae3e_row9_col0, #T_bae3e_row10_col0, #T_bae3e_row12_col0, #T_bae3e_row16_col0, #T_bae3e_row18_col0 {\n",
       "  background-color: steelblue;\n",
       "}\n",
       "#T_bae3e_row1_col4, #T_bae3e_row2_col4, #T_bae3e_row3_col4, #T_bae3e_row5_col4, #T_bae3e_row10_col4, #T_bae3e_row11_col4, #T_bae3e_row13_col4, #T_bae3e_row16_col4, #T_bae3e_row17_col4, #T_bae3e_row20_col4 {\n",
       "  background-color: yellow;\n",
       "}\n",
       "#T_bae3e_row1_col5, #T_bae3e_row4_col5, #T_bae3e_row6_col5, #T_bae3e_row7_col5, #T_bae3e_row10_col5, #T_bae3e_row11_col5, #T_bae3e_row15_col5, #T_bae3e_row16_col5, #T_bae3e_row18_col5, #T_bae3e_row19_col5 {\n",
       "  background-color: blue;\n",
       "}\n",
       "#T_bae3e_row1_col6, #T_bae3e_row2_col6, #T_bae3e_row4_col6, #T_bae3e_row7_col6, #T_bae3e_row8_col6, #T_bae3e_row13_col6, #T_bae3e_row14_col6, #T_bae3e_row17_col6, #T_bae3e_row18_col6 {\n",
       "  background-color: darkgreen;\n",
       "}\n",
       "#T_bae3e_row8_col3, #T_bae3e_row15_col3, #T_bae3e_row19_col3 {\n",
       "  background-color: limegreen;\n",
       "}\n",
       "</style>\n",
       "<table id=\"T_bae3e\">\n",
       "  <thead>\n",
       "    <tr>\n",
       "      <th class=\"blank level0\" >&nbsp;</th>\n",
       "      <th id=\"T_bae3e_level0_col0\" class=\"col_heading level0 col0\" >NM</th>\n",
       "      <th id=\"T_bae3e_level0_col1\" class=\"col_heading level0 col1\" >MP</th>\n",
       "      <th id=\"T_bae3e_level0_col2\" class=\"col_heading level0 col2\" >SA</th>\n",
       "      <th id=\"T_bae3e_level0_col3\" class=\"col_heading level0 col3\" >LKC</th>\n",
       "      <th id=\"T_bae3e_level0_col4\" class=\"col_heading level0 col4\" >DG</th>\n",
       "      <th id=\"T_bae3e_level0_col5\" class=\"col_heading level0 col5\" >MC</th>\n",
       "      <th id=\"T_bae3e_level0_col6\" class=\"col_heading level0 col6\" >AK</th>\n",
       "      <th id=\"T_bae3e_level0_col7\" class=\"col_heading level0 col7\" >CC</th>\n",
       "      <th id=\"T_bae3e_level0_col8\" class=\"col_heading level0 col8\" >KB</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row0\" class=\"row_heading level0 row0\" >0</th>\n",
       "      <td id=\"T_bae3e_row0_col0\" class=\"data row0 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row0_col1\" class=\"data row0 col1\" >B</td>\n",
       "      <td id=\"T_bae3e_row0_col2\" class=\"data row0 col2\" >N</td>\n",
       "      <td id=\"T_bae3e_row0_col3\" class=\"data row0 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row0_col4\" class=\"data row0 col4\" ></td>\n",
       "      <td id=\"T_bae3e_row0_col5\" class=\"data row0 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row0_col6\" class=\"data row0 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row0_col7\" class=\"data row0 col7\" >B</td>\n",
       "      <td id=\"T_bae3e_row0_col8\" class=\"data row0 col8\" >N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row1\" class=\"row_heading level0 row1\" >1</th>\n",
       "      <td id=\"T_bae3e_row1_col0\" class=\"data row1 col0\" >B</td>\n",
       "      <td id=\"T_bae3e_row1_col1\" class=\"data row1 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row1_col2\" class=\"data row1 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row1_col3\" class=\"data row1 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row1_col4\" class=\"data row1 col4\" >N</td>\n",
       "      <td id=\"T_bae3e_row1_col5\" class=\"data row1 col5\" >B</td>\n",
       "      <td id=\"T_bae3e_row1_col6\" class=\"data row1 col6\" >N</td>\n",
       "      <td id=\"T_bae3e_row1_col7\" class=\"data row1 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row1_col8\" class=\"data row1 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row2\" class=\"row_heading level0 row2\" >2</th>\n",
       "      <td id=\"T_bae3e_row2_col0\" class=\"data row2 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row2_col1\" class=\"data row2 col1\" >N</td>\n",
       "      <td id=\"T_bae3e_row2_col2\" class=\"data row2 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row2_col3\" class=\"data row2 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row2_col4\" class=\"data row2 col4\" >B</td>\n",
       "      <td id=\"T_bae3e_row2_col5\" class=\"data row2 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row2_col6\" class=\"data row2 col6\" >B</td>\n",
       "      <td id=\"T_bae3e_row2_col7\" class=\"data row2 col7\" >N</td>\n",
       "      <td id=\"T_bae3e_row2_col8\" class=\"data row2 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row3\" class=\"row_heading level0 row3\" >3</th>\n",
       "      <td id=\"T_bae3e_row3_col0\" class=\"data row3 col0\" >N</td>\n",
       "      <td id=\"T_bae3e_row3_col1\" class=\"data row3 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row3_col2\" class=\"data row3 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row3_col3\" class=\"data row3 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row3_col4\" class=\"data row3 col4\" >B</td>\n",
       "      <td id=\"T_bae3e_row3_col5\" class=\"data row3 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row3_col6\" class=\"data row3 col6\" ></td>\n",
       "      <td id=\"T_bae3e_row3_col7\" class=\"data row3 col7\" >B</td>\n",
       "      <td id=\"T_bae3e_row3_col8\" class=\"data row3 col8\" >N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row4\" class=\"row_heading level0 row4\" >4</th>\n",
       "      <td id=\"T_bae3e_row4_col0\" class=\"data row4 col0\" >B</td>\n",
       "      <td id=\"T_bae3e_row4_col1\" class=\"data row4 col1\" >N</td>\n",
       "      <td id=\"T_bae3e_row4_col2\" class=\"data row4 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row4_col3\" class=\"data row4 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row4_col4\" class=\"data row4 col4\" ></td>\n",
       "      <td id=\"T_bae3e_row4_col5\" class=\"data row4 col5\" >N</td>\n",
       "      <td id=\"T_bae3e_row4_col6\" class=\"data row4 col6\" >B</td>\n",
       "      <td id=\"T_bae3e_row4_col7\" class=\"data row4 col7\" >-</td>\n",
       "      <td id=\"T_bae3e_row4_col8\" class=\"data row4 col8\" ></td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row5\" class=\"row_heading level0 row5\" >5</th>\n",
       "      <td id=\"T_bae3e_row5_col0\" class=\"data row5 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row5_col1\" class=\"data row5 col1\" >B</td>\n",
       "      <td id=\"T_bae3e_row5_col2\" class=\"data row5 col2\" ></td>\n",
       "      <td id=\"T_bae3e_row5_col3\" class=\"data row5 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row5_col4\" class=\"data row5 col4\" >N</td>\n",
       "      <td id=\"T_bae3e_row5_col5\" class=\"data row5 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row5_col6\" class=\"data row5 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row5_col7\" class=\"data row5 col7\" >-</td>\n",
       "      <td id=\"T_bae3e_row5_col8\" class=\"data row5 col8\" >N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row6\" class=\"row_heading level0 row6\" >6</th>\n",
       "      <td id=\"T_bae3e_row6_col0\" class=\"data row6 col0\" >B</td>\n",
       "      <td id=\"T_bae3e_row6_col1\" class=\"data row6 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row6_col2\" class=\"data row6 col2\" >N</td>\n",
       "      <td id=\"T_bae3e_row6_col3\" class=\"data row6 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row6_col4\" class=\"data row6 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row6_col5\" class=\"data row6 col5\" >N</td>\n",
       "      <td id=\"T_bae3e_row6_col6\" class=\"data row6 col6\" ></td>\n",
       "      <td id=\"T_bae3e_row6_col7\" class=\"data row6 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row6_col8\" class=\"data row6 col8\" >B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row7\" class=\"row_heading level0 row7\" >7</th>\n",
       "      <td id=\"T_bae3e_row7_col0\" class=\"data row7 col0\" >N</td>\n",
       "      <td id=\"T_bae3e_row7_col1\" class=\"data row7 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row7_col2\" class=\"data row7 col2\" >B</td>\n",
       "      <td id=\"T_bae3e_row7_col3\" class=\"data row7 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row7_col4\" class=\"data row7 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row7_col5\" class=\"data row7 col5\" >B</td>\n",
       "      <td id=\"T_bae3e_row7_col6\" class=\"data row7 col6\" >N</td>\n",
       "      <td id=\"T_bae3e_row7_col7\" class=\"data row7 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row7_col8\" class=\"data row7 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row8\" class=\"row_heading level0 row8\" >8</th>\n",
       "      <td id=\"T_bae3e_row8_col0\" class=\"data row8 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row8_col1\" class=\"data row8 col1\" >N</td>\n",
       "      <td id=\"T_bae3e_row8_col2\" class=\"data row8 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row8_col3\" class=\"data row8 col3\" >B</td>\n",
       "      <td id=\"T_bae3e_row8_col4\" class=\"data row8 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row8_col5\" class=\"data row8 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row8_col6\" class=\"data row8 col6\" >B</td>\n",
       "      <td id=\"T_bae3e_row8_col7\" class=\"data row8 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row8_col8\" class=\"data row8 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row9\" class=\"row_heading level0 row9\" >9</th>\n",
       "      <td id=\"T_bae3e_row9_col0\" class=\"data row9 col0\" >N</td>\n",
       "      <td id=\"T_bae3e_row9_col1\" class=\"data row9 col1\" >B</td>\n",
       "      <td id=\"T_bae3e_row9_col2\" class=\"data row9 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row9_col3\" class=\"data row9 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row9_col4\" class=\"data row9 col4\" ></td>\n",
       "      <td id=\"T_bae3e_row9_col5\" class=\"data row9 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row9_col6\" class=\"data row9 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row9_col7\" class=\"data row9 col7\" >N</td>\n",
       "      <td id=\"T_bae3e_row9_col8\" class=\"data row9 col8\" >B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row10\" class=\"row_heading level0 row10\" >10</th>\n",
       "      <td id=\"T_bae3e_row10_col0\" class=\"data row10 col0\" >B</td>\n",
       "      <td id=\"T_bae3e_row10_col1\" class=\"data row10 col1\" ></td>\n",
       "      <td id=\"T_bae3e_row10_col2\" class=\"data row10 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row10_col3\" class=\"data row10 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row10_col4\" class=\"data row10 col4\" >N</td>\n",
       "      <td id=\"T_bae3e_row10_col5\" class=\"data row10 col5\" >N</td>\n",
       "      <td id=\"T_bae3e_row10_col6\" class=\"data row10 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row10_col7\" class=\"data row10 col7\" >B</td>\n",
       "      <td id=\"T_bae3e_row10_col8\" class=\"data row10 col8\" ></td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row11\" class=\"row_heading level0 row11\" >11</th>\n",
       "      <td id=\"T_bae3e_row11_col0\" class=\"data row11 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row11_col1\" class=\"data row11 col1\" >N</td>\n",
       "      <td id=\"T_bae3e_row11_col2\" class=\"data row11 col2\" >-</td>\n",
       "      <td id=\"T_bae3e_row11_col3\" class=\"data row11 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row11_col4\" class=\"data row11 col4\" >B</td>\n",
       "      <td id=\"T_bae3e_row11_col5\" class=\"data row11 col5\" >B</td>\n",
       "      <td id=\"T_bae3e_row11_col6\" class=\"data row11 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row11_col7\" class=\"data row11 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row11_col8\" class=\"data row11 col8\" >N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row12\" class=\"row_heading level0 row12\" >12</th>\n",
       "      <td id=\"T_bae3e_row12_col0\" class=\"data row12 col0\" >N</td>\n",
       "      <td id=\"T_bae3e_row12_col1\" class=\"data row12 col1\" ></td>\n",
       "      <td id=\"T_bae3e_row12_col2\" class=\"data row12 col2\" >B</td>\n",
       "      <td id=\"T_bae3e_row12_col3\" class=\"data row12 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row12_col4\" class=\"data row12 col4\" ></td>\n",
       "      <td id=\"T_bae3e_row12_col5\" class=\"data row12 col5\" >-</td>\n",
       "      <td id=\"T_bae3e_row12_col6\" class=\"data row12 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row12_col7\" class=\"data row12 col7\" >B</td>\n",
       "      <td id=\"T_bae3e_row12_col8\" class=\"data row12 col8\" ></td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row13\" class=\"row_heading level0 row13\" >13</th>\n",
       "      <td id=\"T_bae3e_row13_col0\" class=\"data row13 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row13_col1\" class=\"data row13 col1\" >N</td>\n",
       "      <td id=\"T_bae3e_row13_col2\" class=\"data row13 col2\" ></td>\n",
       "      <td id=\"T_bae3e_row13_col3\" class=\"data row13 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row13_col4\" class=\"data row13 col4\" >B</td>\n",
       "      <td id=\"T_bae3e_row13_col5\" class=\"data row13 col5\" >-</td>\n",
       "      <td id=\"T_bae3e_row13_col6\" class=\"data row13 col6\" >N</td>\n",
       "      <td id=\"T_bae3e_row13_col7\" class=\"data row13 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row13_col8\" class=\"data row13 col8\" >B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row14\" class=\"row_heading level0 row14\" >14</th>\n",
       "      <td id=\"T_bae3e_row14_col0\" class=\"data row14 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row14_col1\" class=\"data row14 col1\" >B</td>\n",
       "      <td id=\"T_bae3e_row14_col2\" class=\"data row14 col2\" >N</td>\n",
       "      <td id=\"T_bae3e_row14_col3\" class=\"data row14 col3\" ></td>\n",
       "      <td id=\"T_bae3e_row14_col4\" class=\"data row14 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row14_col5\" class=\"data row14 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row14_col6\" class=\"data row14 col6\" >B</td>\n",
       "      <td id=\"T_bae3e_row14_col7\" class=\"data row14 col7\" >N</td>\n",
       "      <td id=\"T_bae3e_row14_col8\" class=\"data row14 col8\" ></td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row15\" class=\"row_heading level0 row15\" >15</th>\n",
       "      <td id=\"T_bae3e_row15_col0\" class=\"data row15 col0\" >-</td>\n",
       "      <td id=\"T_bae3e_row15_col1\" class=\"data row15 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row15_col2\" class=\"data row15 col2\" >B</td>\n",
       "      <td id=\"T_bae3e_row15_col3\" class=\"data row15 col3\" >N</td>\n",
       "      <td id=\"T_bae3e_row15_col4\" class=\"data row15 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row15_col5\" class=\"data row15 col5\" >N</td>\n",
       "      <td id=\"T_bae3e_row15_col6\" class=\"data row15 col6\" ></td>\n",
       "      <td id=\"T_bae3e_row15_col7\" class=\"data row15 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row15_col8\" class=\"data row15 col8\" >B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row16\" class=\"row_heading level0 row16\" >16</th>\n",
       "      <td id=\"T_bae3e_row16_col0\" class=\"data row16 col0\" >B</td>\n",
       "      <td id=\"T_bae3e_row16_col1\" class=\"data row16 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row16_col2\" class=\"data row16 col2\" ></td>\n",
       "      <td id=\"T_bae3e_row16_col3\" class=\"data row16 col3\" ></td>\n",
       "      <td id=\"T_bae3e_row16_col4\" class=\"data row16 col4\" >N</td>\n",
       "      <td id=\"T_bae3e_row16_col5\" class=\"data row16 col5\" >B</td>\n",
       "      <td id=\"T_bae3e_row16_col6\" class=\"data row16 col6\" ></td>\n",
       "      <td id=\"T_bae3e_row16_col7\" class=\"data row16 col7\" >N</td>\n",
       "      <td id=\"T_bae3e_row16_col8\" class=\"data row16 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row17\" class=\"row_heading level0 row17\" >17</th>\n",
       "      <td id=\"T_bae3e_row17_col0\" class=\"data row17 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row17_col1\" class=\"data row17 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row17_col2\" class=\"data row17 col2\" >N</td>\n",
       "      <td id=\"T_bae3e_row17_col3\" class=\"data row17 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row17_col4\" class=\"data row17 col4\" >B</td>\n",
       "      <td id=\"T_bae3e_row17_col5\" class=\"data row17 col5\" ></td>\n",
       "      <td id=\"T_bae3e_row17_col6\" class=\"data row17 col6\" >N</td>\n",
       "      <td id=\"T_bae3e_row17_col7\" class=\"data row17 col7\" >B</td>\n",
       "      <td id=\"T_bae3e_row17_col8\" class=\"data row17 col8\" >-</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row18\" class=\"row_heading level0 row18\" >18</th>\n",
       "      <td id=\"T_bae3e_row18_col0\" class=\"data row18 col0\" >N</td>\n",
       "      <td id=\"T_bae3e_row18_col1\" class=\"data row18 col1\" >-</td>\n",
       "      <td id=\"T_bae3e_row18_col2\" class=\"data row18 col2\" >B</td>\n",
       "      <td id=\"T_bae3e_row18_col3\" class=\"data row18 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row18_col4\" class=\"data row18 col4\" ></td>\n",
       "      <td id=\"T_bae3e_row18_col5\" class=\"data row18 col5\" >N</td>\n",
       "      <td id=\"T_bae3e_row18_col6\" class=\"data row18 col6\" >B</td>\n",
       "      <td id=\"T_bae3e_row18_col7\" class=\"data row18 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row18_col8\" class=\"data row18 col8\" ></td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row19\" class=\"row_heading level0 row19\" >19</th>\n",
       "      <td id=\"T_bae3e_row19_col0\" class=\"data row19 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row19_col1\" class=\"data row19 col1\" >B</td>\n",
       "      <td id=\"T_bae3e_row19_col2\" class=\"data row19 col2\" ></td>\n",
       "      <td id=\"T_bae3e_row19_col3\" class=\"data row19 col3\" >N</td>\n",
       "      <td id=\"T_bae3e_row19_col4\" class=\"data row19 col4\" >-</td>\n",
       "      <td id=\"T_bae3e_row19_col5\" class=\"data row19 col5\" >B</td>\n",
       "      <td id=\"T_bae3e_row19_col6\" class=\"data row19 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row19_col7\" class=\"data row19 col7\" ></td>\n",
       "      <td id=\"T_bae3e_row19_col8\" class=\"data row19 col8\" >N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th id=\"T_bae3e_level0_row20\" class=\"row_heading level0 row20\" >20</th>\n",
       "      <td id=\"T_bae3e_row20_col0\" class=\"data row20 col0\" ></td>\n",
       "      <td id=\"T_bae3e_row20_col1\" class=\"data row20 col1\" ></td>\n",
       "      <td id=\"T_bae3e_row20_col2\" class=\"data row20 col2\" >B</td>\n",
       "      <td id=\"T_bae3e_row20_col3\" class=\"data row20 col3\" >-</td>\n",
       "      <td id=\"T_bae3e_row20_col4\" class=\"data row20 col4\" >N</td>\n",
       "      <td id=\"T_bae3e_row20_col5\" class=\"data row20 col5\" >-</td>\n",
       "      <td id=\"T_bae3e_row20_col6\" class=\"data row20 col6\" >-</td>\n",
       "      <td id=\"T_bae3e_row20_col7\" class=\"data row20 col7\" >N</td>\n",
       "      <td id=\"T_bae3e_row20_col8\" class=\"data row20 col8\" >B</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n"
      ],
      "text/plain": [
       "<pandas.io.formats.style.Styler at 0x7fb9835b32e0>"
      ]
     },
     "execution_count": 125,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Import data\n",
    "avail = pd.read_csv('availability.csv')\n",
    "staff_data = pd.read_csv('staff.csv')\n",
    "names = avail.T.index\n",
    "staff = staff_data[staff_data.person == names]\n",
    "\n",
    "# Sets\n",
    "num_weeks, num_people = avail.shape\n",
    "people = range(0,num_people)\n",
    "weeks = range(0,num_weeks)\n",
    "is_resident = staff.resident\n",
    "res = is_resident[is_resident == 1].index\n",
    "path = is_resident[is_resident == 0].index\n",
    "\n",
    "# Paramaters\n",
    "n_target = (staff.n_target * num_weeks).to_numpy()\n",
    "b_target = (staff.b_target * num_weeks).to_numpy()\n",
    "\n",
    "n_rp_freq = n_target[res]/len(path)\n",
    "n_rp_target = np.tile(n_rp_freq.reshape((-1,1)), len(path))\n",
    "\n",
    "b_rp_freq = b_target[res]/len(path)\n",
    "b_rp_target = np.tile(b_rp_freq.reshape((-1,1)), len(path))\n",
    "\n",
    "# Decision variables\n",
    "n = cp.Variable(avail.shape, boolean=True, name='necropsies')\n",
    "b = cp.Variable(avail.shape, boolean=True, name='biopsies')\n",
    "\n",
    "# Definition variables\n",
    "c = n + b\n",
    "constraints = []\n",
    "\n",
    "# Constraint: Each duty requires one pathologist\n",
    "c_one_path_n = [cp.sum(n[w,path]) == 1 for w in weeks]\n",
    "c_one_path_b = [cp.sum(b[w,path]) == 1 for w in weeks]\n",
    "constraints += c_one_path_n + c_one_path_b\n",
    "\n",
    "# Constraint: Each duty can have a max of one resident\n",
    "c_one_res_n = [cp.sum(n[w,res]) <= 1 for w in weeks]\n",
    "c_one_res_b = [cp.sum(b[w,res]) <= 1 for w in weeks]\n",
    "constraints += c_one_res_n + c_one_res_b\n",
    "\n",
    "# Constraint: Cannot work both duties or during time off\n",
    "c_work_avail = [c[w,p] <= avail.iloc[w,p] for w in weeks for p in people]\n",
    "constraints += c_work_avail\n",
    "\n",
    "# Constraint: Residents needing RARC must have 2 consecutive weeks off at least once\n",
    "\n",
    "# Cost: Penalize three weeks off?\n",
    "\n",
    "# Cost: Penalize two weeks on followed by unavailable week -> depends on type of unavailability?\n",
    "\n",
    "# Cost: Two consecutive duties excluding NB -> refactor using c variable and reward nb\n",
    "nn_first = cp.Variable(avail.shape, boolean=True)\n",
    "nn_second = cp.Variable(avail.shape, boolean=True)\n",
    "bb_first = cp.Variable(avail.shape, boolean=True)\n",
    "bb_second = cp.Variable(avail.shape, boolean=True)\n",
    "bn_first = cp.Variable(avail.shape, boolean=True)\n",
    "bn_second = cp.Variable(avail.shape, boolean=True)\n",
    "\n",
    "c_nn_doubles = [nn_first[w] + nn_second[w] == cp.sum(n[w:w+2], axis=0) for w in weeks]\n",
    "c_bb_doubles = [bb_first[w] + bb_second[w] == cp.sum(b[w:w+2], axis=0) for w in weeks]\n",
    "c_bn_doubles = [bn_first[w] + bn_second[w] == b[w] + n[w+1] for w in weeks[:-1]]\n",
    "constraints += c_nn_doubles + c_bb_doubles + c_bn_doubles\n",
    "\n",
    "# Cost: Three consecutive duties\n",
    "triple_one_two = cp.Variable(avail.shape, integer=True)\n",
    "triple_third = cp.Variable(avail.shape, boolean=True)\n",
    "c_triples = [triple_one_two[w] + triple_third[w] == cp.sum(c[w:w+3], axis=0) for w in weeks]\n",
    "constraints += c_triples + [triple_one_two <= 2]\n",
    "\n",
    "# Cost: Absolute difference between target Bx and scheduled Bx\n",
    "b_diff = cp.Variable(num_people)\n",
    "b_below = cp.Variable(num_people, nonneg=True)\n",
    "b_above = cp.Variable(num_people, nonneg=True)\n",
    "\n",
    "c_b_diff = [b_diff == cp.sum(b, axis=0) - b_target]\n",
    "c_b_below = [b_below >= cp.neg(b_diff)]\n",
    "c_b_above = [b_above >= b_diff]\n",
    "constraints += c_b_diff + c_b_below + c_b_above\n",
    "\n",
    "# Cost: Absolute difference between target Nx and scheduled Nx\n",
    "n_diff = cp.Variable(num_people)\n",
    "n_below = cp.Variable(num_people, nonneg=True)\n",
    "n_above = cp.Variable(num_people, nonneg=True)\n",
    "\n",
    "c_n_diff = [n_diff == cp.sum(n, axis=0) - n_target]\n",
    "c_n_below = [n_below >= cp.neg(n_diff)]\n",
    "c_n_above = [n_above >= n_diff]\n",
    "constraints += c_n_diff + c_n_below + c_n_above\n",
    "\n",
    "# Cost: Shortage of target Nx R/P pairs\n",
    "n_pairs = [cp.Variable((len(res), len(path)), boolean=True) for _ in weeks]\n",
    "n_pairs_below = cp.Variable((len(res), len(path)), nonneg=True)\n",
    "\n",
    "c1_n_pairs = [n_pairs[w][ir,ip] <= n[w,p] for ip,p in enumerate(path) for ir,_ in enumerate(res) for w in weeks]\n",
    "c2_n_pairs =[n_pairs[w][ir,ip] <= n[w,r] for ip,_ in enumerate(path) for ir,r in enumerate(res) for w in weeks]\n",
    "c3_n_pairs = [n_pairs_below >= n_rp_target - cp.sum(n_pairs, axis=2)]\n",
    "constraints += c1_n_pairs + c2_n_pairs + c3_n_pairs\n",
    "\n",
    "# Cost: Shortage of target Bx R/P pairs\n",
    "b_pairs = [cp.Variable((len(res), len(path)), boolean=True) for _ in weeks]\n",
    "b_pairs_below = cp.Variable((len(res), len(path)), nonneg=True)\n",
    "\n",
    "c1_b_pairs = [b_pairs[w][ir,ip] <= b[w,p] for ip,p in enumerate(path) for ir,_ in enumerate(res) for w in weeks]\n",
    "c2_b_pairs =[b_pairs[w][ir,ip] <= b[w,r] for ip,_ in enumerate(path) for ir,r in enumerate(res) for w in weeks]\n",
    "c3_b_pairs = [b_pairs_below >= b_rp_target - cp.sum(b_pairs, axis=2)]\n",
    "constraints += c1_b_pairs + c2_b_pairs + c3_b_pairs\n",
    "\n",
    "# Define costs\n",
    "double_cost = 3*cp.sum(nn_second) + 2*cp.sum(bn_second) + cp.sum(bb_second)\n",
    "triple_cost = cp.sum(triple_third)\n",
    "duty_cost = cp.sum(b_below + b_above + n_below + n_above)\n",
    "pair_cost = cp.sum(n_pairs_below + b_pairs_below)\n",
    "\n",
    "# Solve\n",
    "obj = cp.Minimize(double_cost + triple_cost + duty_cost + pair_cost)\n",
    "prob = cp.Problem(obj, constraints)\n",
    "prob.solve(solver='GUROBI')\n",
    "viz_schedule(avail, b.value, n.value)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "40d3a090f54c6569ab1632332b64b2c03c39dcf918b08424e98f38b5ae0af88f"
  },
  "kernelspec": {
   "display_name": "Python 3.8.8 ('base')",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.7"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}