courseplannerFIN.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "ef2a326a",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import math"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "b3a92508",
   "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>Class</th>\n",
       "      <th>Title</th>\n",
       "      <th>Units</th>\n",
       "      <th>Prereq</th>\n",
       "      <th>Coreq</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>ADJUS 120</td>\n",
       "      <td>Introduction to the Administration of Justice</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>ADJUS 121</td>\n",
       "      <td>Criminal Law</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>ADJUS 126</td>\n",
       "      <td>Leadership and Ethics</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>ADJUS 130</td>\n",
       "      <td>Cultural Diversity in Criminal Justice</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>ANTHR 115</td>\n",
       "      <td>Primate Evolution and Adaptation</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>ANTHR 120</td>\n",
       "      <td>Magic, Witchcraft, and Religion in the Americas</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>ANTHR 125</td>\n",
       "      <td>Introduction to Archaeology and Prehistory</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>ANTHR 130</td>\n",
       "      <td>Cultural Anthropology</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>ANTHR 135</td>\n",
       "      <td>Native Americans</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>ANTHR 140</td>\n",
       "      <td>Biological Anthropology</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>ANTHR 141L</td>\n",
       "      <td>Biological Anthropology Laboratory</td>\n",
       "      <td>1</td>\n",
       "      <td>N</td>\n",
       "      <td>ANTHR 140</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>ARABC 120</td>\n",
       "      <td>First Term Arabic</td>\n",
       "      <td>5</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>ARABC 121</td>\n",
       "      <td>Second Term Arabic</td>\n",
       "      <td>5</td>\n",
       "      <td>ARABC 120</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                        Class  \\\n",
       "0   Administration of Justice   \n",
       "1                   ADJUS 120   \n",
       "2                   ADJUS 121   \n",
       "3                   ADJUS 126   \n",
       "4                   ADJUS 130   \n",
       "5                         NaN   \n",
       "6                Anthropology   \n",
       "7                   ANTHR 115   \n",
       "8                   ANTHR 120   \n",
       "9                   ANTHR 125   \n",
       "10                  ANTHR 130   \n",
       "11                  ANTHR 135   \n",
       "12                  ANTHR 140   \n",
       "13                 ANTHR 141L   \n",
       "14                        NaN   \n",
       "15                     Arabic   \n",
       "16                  ARABC 120   \n",
       "17                  ARABC 121   \n",
       "18                        NaN   \n",
       "19               Architecture   \n",
       "\n",
       "                                              Title  \\\n",
       "0                         Administration of Justice   \n",
       "1     Introduction to the Administration of Justice   \n",
       "2                                      Criminal Law   \n",
       "3                             Leadership and Ethics   \n",
       "4            Cultural Diversity in Criminal Justice   \n",
       "5                                               NaN   \n",
       "6                                      Anthropology   \n",
       "7                  Primate Evolution and Adaptation   \n",
       "8   Magic, Witchcraft, and Religion in the Americas   \n",
       "9        Introduction to Archaeology and Prehistory   \n",
       "10                            Cultural Anthropology   \n",
       "11                                 Native Americans   \n",
       "12                          Biological Anthropology   \n",
       "13               Biological Anthropology Laboratory   \n",
       "14                                              NaN   \n",
       "15                                           Arabic   \n",
       "16                                First Term Arabic   \n",
       "17                               Second Term Arabic   \n",
       "18                                              NaN   \n",
       "19                                     Architecture   \n",
       "\n",
       "                        Units                     Prereq  \\\n",
       "0   Administration of Justice  Administration of Justice   \n",
       "1                           3                          N   \n",
       "2                           3                          N   \n",
       "3                           3                          N   \n",
       "4                           3                          N   \n",
       "5                         NaN                        NaN   \n",
       "6                Anthropology               Anthropology   \n",
       "7                           3                          N   \n",
       "8                           3                          N   \n",
       "9                           3                          N   \n",
       "10                          3                          N   \n",
       "11                          3                          N   \n",
       "12                          3                          N   \n",
       "13                          1                          N   \n",
       "14                        NaN                        NaN   \n",
       "15                     Arabic                     Arabic   \n",
       "16                          5                          N   \n",
       "17                          5                  ARABC 120   \n",
       "18                        NaN                        NaN   \n",
       "19               Architecture               Architecture   \n",
       "\n",
       "                        Coreq  \n",
       "0   Administration of Justice  \n",
       "1                           N  \n",
       "2                           N  \n",
       "3                           N  \n",
       "4                           N  \n",
       "5                         NaN  \n",
       "6                Anthropology  \n",
       "7                           N  \n",
       "8                           N  \n",
       "9                           N  \n",
       "10                          N  \n",
       "11                          N  \n",
       "12                          N  \n",
       "13                  ANTHR 140  \n",
       "14                        NaN  \n",
       "15                     Arabic  \n",
       "16                          N  \n",
       "17                          N  \n",
       "18                        NaN  \n",
       "19               Architecture  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = pd.read_csv('CleanedDVCwCoreqsV3.csv')\n",
    "df.head(20)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "45a9fc7e",
   "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>Title</th>\n",
       "      <th>Units</th>\n",
       "      <th>Prereq</th>\n",
       "      <th>Coreq</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Class</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Administration of Justice</th>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "      <td>Administration of Justice</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ADJUS 120</th>\n",
       "      <td>Introduction to the Administration of Justice</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ADJUS 121</th>\n",
       "      <td>Criminal Law</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ADJUS 126</th>\n",
       "      <td>Leadership and Ethics</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ADJUS 130</th>\n",
       "      <td>Cultural Diversity in Criminal Justice</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Anthropology</th>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "      <td>Anthropology</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 115</th>\n",
       "      <td>Primate Evolution and Adaptation</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 120</th>\n",
       "      <td>Magic, Witchcraft, and Religion in the Americas</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 125</th>\n",
       "      <td>Introduction to Archaeology and Prehistory</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 130</th>\n",
       "      <td>Cultural Anthropology</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 135</th>\n",
       "      <td>Native Americans</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 140</th>\n",
       "      <td>Biological Anthropology</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ANTHR 141L</th>\n",
       "      <td>Biological Anthropology Laboratory</td>\n",
       "      <td>1</td>\n",
       "      <td>N</td>\n",
       "      <td>ANTHR 140</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Arabic</th>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "      <td>Arabic</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ARABC 120</th>\n",
       "      <td>First Term Arabic</td>\n",
       "      <td>5</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ARABC 121</th>\n",
       "      <td>Second Term Arabic</td>\n",
       "      <td>5</td>\n",
       "      <td>ARABC 120</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Architecture</th>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "      <td>Architecture</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ARCHI 119</th>\n",
       "      <td>Introduction to Technical Drawing</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ARCHI 120</th>\n",
       "      <td>Introduction to Architecture and Environmental...</td>\n",
       "      <td>3</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ARCHI 121</th>\n",
       "      <td>Architectural Design I</td>\n",
       "      <td>4</td>\n",
       "      <td>ARCHI 120</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                                       Title  \\\n",
       "Class                                                                          \n",
       "Administration of Justice                          Administration of Justice   \n",
       "ADJUS 120                      Introduction to the Administration of Justice   \n",
       "ADJUS 121                                                       Criminal Law   \n",
       "ADJUS 126                                              Leadership and Ethics   \n",
       "ADJUS 130                             Cultural Diversity in Criminal Justice   \n",
       "Anthropology                                                    Anthropology   \n",
       "ANTHR 115                                   Primate Evolution and Adaptation   \n",
       "ANTHR 120                    Magic, Witchcraft, and Religion in the Americas   \n",
       "ANTHR 125                         Introduction to Archaeology and Prehistory   \n",
       "ANTHR 130                                              Cultural Anthropology   \n",
       "ANTHR 135                                                   Native Americans   \n",
       "ANTHR 140                                            Biological Anthropology   \n",
       "ANTHR 141L                                Biological Anthropology Laboratory   \n",
       "Arabic                                                                Arabic   \n",
       "ARABC 120                                                  First Term Arabic   \n",
       "ARABC 121                                                 Second Term Arabic   \n",
       "Architecture                                                    Architecture   \n",
       "ARCHI 119                                  Introduction to Technical Drawing   \n",
       "ARCHI 120                  Introduction to Architecture and Environmental...   \n",
       "ARCHI 121                                             Architectural Design I   \n",
       "\n",
       "                                               Units  \\\n",
       "Class                                                  \n",
       "Administration of Justice  Administration of Justice   \n",
       "ADJUS 120                                          3   \n",
       "ADJUS 121                                          3   \n",
       "ADJUS 126                                          3   \n",
       "ADJUS 130                                          3   \n",
       "Anthropology                            Anthropology   \n",
       "ANTHR 115                                          3   \n",
       "ANTHR 120                                          3   \n",
       "ANTHR 125                                          3   \n",
       "ANTHR 130                                          3   \n",
       "ANTHR 135                                          3   \n",
       "ANTHR 140                                          3   \n",
       "ANTHR 141L                                         1   \n",
       "Arabic                                        Arabic   \n",
       "ARABC 120                                          5   \n",
       "ARABC 121                                          5   \n",
       "Architecture                            Architecture   \n",
       "ARCHI 119                                          3   \n",
       "ARCHI 120                                          3   \n",
       "ARCHI 121                                          4   \n",
       "\n",
       "                                              Prereq  \\\n",
       "Class                                                  \n",
       "Administration of Justice  Administration of Justice   \n",
       "ADJUS 120                                          N   \n",
       "ADJUS 121                                          N   \n",
       "ADJUS 126                                          N   \n",
       "ADJUS 130                                          N   \n",
       "Anthropology                            Anthropology   \n",
       "ANTHR 115                                          N   \n",
       "ANTHR 120                                          N   \n",
       "ANTHR 125                                          N   \n",
       "ANTHR 130                                          N   \n",
       "ANTHR 135                                          N   \n",
       "ANTHR 140                                          N   \n",
       "ANTHR 141L                                         N   \n",
       "Arabic                                        Arabic   \n",
       "ARABC 120                                          N   \n",
       "ARABC 121                                  ARABC 120   \n",
       "Architecture                            Architecture   \n",
       "ARCHI 119                                          N   \n",
       "ARCHI 120                                          N   \n",
       "ARCHI 121                                  ARCHI 120   \n",
       "\n",
       "                                               Coreq  \n",
       "Class                                                 \n",
       "Administration of Justice  Administration of Justice  \n",
       "ADJUS 120                                          N  \n",
       "ADJUS 121                                          N  \n",
       "ADJUS 126                                          N  \n",
       "ADJUS 130                                          N  \n",
       "Anthropology                            Anthropology  \n",
       "ANTHR 115                                          N  \n",
       "ANTHR 120                                          N  \n",
       "ANTHR 125                                          N  \n",
       "ANTHR 130                                          N  \n",
       "ANTHR 135                                          N  \n",
       "ANTHR 140                                          N  \n",
       "ANTHR 141L                                 ANTHR 140  \n",
       "Arabic                                        Arabic  \n",
       "ARABC 120                                          N  \n",
       "ARABC 121                                          N  \n",
       "Architecture                            Architecture  \n",
       "ARCHI 119                                          N  \n",
       "ARCHI 120                                          N  \n",
       "ARCHI 121                                          N  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.dropna(inplace = True) #dropping empty rows between different departments \n",
    "df.set_index('Class', inplace = True) #setting Class col to be index of dataframe\n",
    "df.head(20) #displaying first 20 rows"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "29cabda6",
   "metadata": {},
   "source": [
    "# UC Berkeley"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b94e289d",
   "metadata": {},
   "source": [
    "### UC Berkeley transfer agreements"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "ef453201",
   "metadata": {},
   "outputs": [],
   "source": [
    "aeroSpaceBerk = {\n",
    "    'reqCourses' : ['MATH 192', 'MATH 193', 'MATH 292', 'MATH 194', 'MATH 294', 'PHYS 130', 'PHYS 230', 'ENGL 122', 'CHEM 120' ],\n",
    "    'recCourses' : ['ENGIN 136', 'ENGIN 240', 'ENGIN 257', 'ENGIN 210']\n",
    "    \n",
    "}\n",
    "\n",
    "anthropologyBerk = {\n",
    "    'reqCourses' : ['ANTHR 140', 'ANTHR 125', 'ANTHR 130'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "#corec test phys 130, math 193\n",
    "architectureBerk = {\n",
    "    'reqCourses' : ['ARCHI 120', 'ARCHI 130', 'ARCHI 131', 'ARCHI 121', 'ARCHI 220', 'MATH 192', 'PHYS 130', 'ENGL 122', 'ENGL 126'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "\n",
    "businessAdBerk  = {\n",
    "    'reqCourses' : ['BUS 109', 'ECON 220', 'ECON 221', 'BUS 240', 'MATH 142', 'PSYCH 214', 'MATH 192', 'MATH 193', 'ENGL 122', 'ENGL 126'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "\n",
    "berkeley = {'Aerospace Engineering' : aeroSpaceBerk,\n",
    "            'Anthropology' : anthropologyBerk,\n",
    "            'Architecture' : architectureBerk,\n",
    "            'Business Administration' : businessAdBerk}\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d5acfe1f",
   "metadata": {},
   "source": [
    "# UCLA"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "9d7915d8",
   "metadata": {},
   "source": [
    "#### UCLA Transfer Agreements"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "6785a067",
   "metadata": {},
   "outputs": [],
   "source": [
    "aeroSpaceLA = {\n",
    "    'reqCourses' : ['CHEM 120', 'CHEM 121', 'MATH 192', 'MATH 193', 'MATH 292', 'MATH 194', 'MATH 294', 'PHYS 130', 'PHYS 230', 'PHYS 231', 'ENGL 122', 'ENGL 126', 'COMSC 165'],\n",
    "    'recCourses' : ['COMSC 200', 'ENGIN 136', 'ENGIN 230', 'ENGIN 257']\n",
    "}\n",
    "\n",
    "anthropologyLA = {\n",
    "    'reqCourses' : ['ANTHR 140', 'ANTHR 125', 'ANTHR 130'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "\n",
    "architectureLA = {\n",
    "    'reqCourses' : ['ENGL 122', 'ENGL 126', 'MATH 192'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "\n",
    "businessEconLA = {\n",
    "    'reqCourses' : ['ECON 220', 'ECON 221', 'BUSAC 186', 'BUSAC 187', 'MATH 192', 'MATH 193', 'ENGL 126'],\n",
    "    'recCourses' : []\n",
    "}\n",
    "\n",
    "\n",
    "ucla = {'Aerospace Engineering' : aeroSpaceLA,\n",
    "        'Anthropology' : anthropologyLA,\n",
    "        'Architectural Studies' : architectureLA,\n",
    "        'Business Economics' : businessEconLA}\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "a0311c5b",
   "metadata": {},
   "outputs": [],
   "source": [
    "#gets units of class when provided with abreviation ex: getUnits('ENGL 122') returns 3\n",
    "def getUnits(classAbv):\n",
    "    cl = df.loc[classAbv]  #getting row of class info (series) from dataframe\n",
    "    units = float(cl.Units) #gets Units from row and casts to float\n",
    "    return units"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "b8a58f66",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['PHYS 230']"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#gets a class name as input and returns a list of one or more prereqs (list of strs) DOES NOT INCLUDE COREQ MARKING\n",
    "def getPrereqs(classAbv):\n",
    "    cl = df.loc[classAbv]\n",
    "    prereqs = cl.Prereq.split('-')  #not worrying abt coreqs rn so removing ?s and splitting if mulitple prereqs\n",
    "    return prereqs\n",
    "\n",
    "#------------------------------Example                            \n",
    "test = getPrereqs('PHYS 231')\n",
    "test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "23aaef38",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['PHYS 121', 'MATH 193']"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def getCoreqs(classAbv):\n",
    "    cl = df.loc[classAbv]\n",
    "    coreqs = cl.Coreq.split('-')  #not worrying abt coreqs rn so removing ?s and splitting if mulitple prereqs\n",
    "    return coreqs\n",
    "\n",
    "#------------------------------Example                            \n",
    "test = getCoreqs('PHYS 125')\n",
    "test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "1472c959",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def hasPrereq(classAbv):\n",
    "    ph = getPrereqs(classAbv)\n",
    "    if ph[0] == 'N':\n",
    "        return False\n",
    "    return True\n",
    "\n",
    "#------------------------------Example\n",
    "n = hasPrereq('ENGL 126')\n",
    "n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "c6bf55fd",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def hasCoreq(classAbv):\n",
    "    ph = getCoreqs(classAbv)\n",
    "    if ph[0] == 'N':\n",
    "        return False\n",
    "    return True\n",
    "\n",
    "#--------------------------Example\n",
    "n = hasCoreq('PHYS 125')\n",
    "n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "d148a316",
   "metadata": {},
   "outputs": [],
   "source": [
    "#colleges is list of dictionaries (not strings) for Ex: [berkeley, ucla], majors is a list strings, each being a major name\n",
    "#correspoding to the college in the same position of the college array as the major. Also removes any repeat courses.\n",
    "#colleges = [c1, c2]  majors = ['m1', 'm2'] (input lists must be same leng). Returns a NumPy array.\n",
    "def getStudentRequiredCourses(colleges, majors):\n",
    "    sRq = []\n",
    "    for i, colls in enumerate(colleges):\n",
    "        sRq += colls[majors[i]]['reqCourses']\n",
    "    return np.unique(sRq)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "2ee3b28e",
   "metadata": {},
   "outputs": [],
   "source": [
    "#same as above just for recommended courses\n",
    "def getStudentRecommendedCourses(colleges, majors):\n",
    "    sRc = []\n",
    "    for i, colls in enumerate(colleges):\n",
    "        sRc += colls[majors[i]]['recCourses']\n",
    "    return np.unique(sRc)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "d729e3ba",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['BUSAC 186' 'BUSAC 187' 'ECON 220' 'ECON 221' 'ENGL 126' 'MATH 192'\n",
      " 'MATH 193']\n",
      "new\n",
      "['BUSAC 186' 'BUSAC 187' 'ECON 220' 'ECON 221' 'ENGL 122' 'ENGL 126'\n",
      " 'MATH 192' 'MATH 193']\n"
     ]
    }
   ],
   "source": [
    "#input is list of courses, returns new list of courses (maybe be unaltered though)\n",
    "#function adds prereqs that arent in required courses list that need to be\n",
    "def addPrereqs(courses):\n",
    "    newCourses = courses\n",
    "    for course in courses:  \n",
    "        ph = getPrereqs(course)   #gets prereq list of a course\n",
    "        if ph[0] == 'N':          #if list contains 'N' it means no prereqs\n",
    "            continue\n",
    "        else:                    #else go through list of prereqs, check if they are in the list of all courses, if not append to prereq to the list of all courses\n",
    "            for prereq in ph:\n",
    "                if prereq not in newCourses:\n",
    "                    newCourses = np.append(newCourses, prereq)    \n",
    "    return np.sort(newCourses)            #not sure if sorting here is useful yet (maybe take out later?)\n",
    "\n",
    "#------------------------------------------- Example (of this and previous two functions)\n",
    "cou = [ucla]\n",
    "maj = ['Business Economics']\n",
    "sC = getStudentRequiredCourses(cou, maj)\n",
    "print(sC)\n",
    "nsC = addPrereqs(sC)\n",
    "print('new')\n",
    "print(nsC)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "ab481f67",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['ARCHI 120' 'ARCHI 121' 'ARCHI 130' 'ARCHI 131' 'ARCHI 220' 'ENGL 122'\n",
      " 'ENGL 126' 'MATH 192' 'PHYS 130']\n",
      "new\n",
      "['ARCHI 120' 'ARCHI 121' 'ARCHI 130' 'ARCHI 131' 'ARCHI 220' 'ENGL 122'\n",
      " 'ENGL 126' 'MATH 192' 'MATH 193' 'PHYS 130']\n"
     ]
    }
   ],
   "source": [
    "def addCoreqs(courses):\n",
    "    newCourses = courses\n",
    "    for course in courses:  \n",
    "        ph = getCoreqs(course)   #gets coreq list of a course\n",
    "        if ph[0] == 'N':          #if list contains 'N' it means no coreqs\n",
    "            continue\n",
    "        else:                    #else go through list of coreqs, check if they are in the list of all courses, if not append to the list of all courses\n",
    "            for coreq in ph:\n",
    "                if coreq not in newCourses:\n",
    "                    newCourses = np.append(newCourses, coreq)    \n",
    "    return np.sort(newCourses)            #sorting to make sure order is kept\n",
    "\n",
    "#---------------------------------------------------- example\n",
    "cul = [berkeley]\n",
    "mij = ['Architecture']\n",
    "sp = getStudentRequiredCourses(cul, mij)\n",
    "print(sp)\n",
    "nsp = addCoreqs(sp)\n",
    "print('new')\n",
    "print(nsp)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "9988a699",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'ARCHI': array(['ARCHI 120', 'ARCHI 121', 'ARCHI 130', 'ARCHI 131', 'ARCHI 220'],\n",
       "       dtype='<U9'),\n",
       " 'ENGL': array(['ENGL 122', 'ENGL 126'], dtype='<U8'),\n",
       " 'MATH': array(['MATH 192'], dtype='<U8'),\n",
       " 'PHYS': array(['PHYS 130'], dtype='<U8')}"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#input is list/arr of strs each being a course name, returns a dictionary of courses sorted by their deparment ex: {'BUS': ['BUS 109', 'BUS 110'], etc}\n",
    "def splitCourses(courses):\n",
    "    course_dict = {}\n",
    "    for course in courses:\n",
    "        abbreviation = course.split()[0]            #split str into list to get abbreviation of course (MATH 240 becomes MATH) to be the keys for course dictionary\n",
    "        if abbreviation in course_dict:             #if the abbreviation (key) is already there you can add the class to the arr corresponding to said key        \n",
    "            course_dict[abbreviation] = np.append(course_dict[abbreviation], course)\n",
    "        else:                                       #if not found create a key pair value with abbreviation and a new arr with the course starting it\n",
    "            course_dict[abbreviation] = np.array([course])\n",
    "\n",
    "    return course_dict\n",
    "\n",
    "#---------------------------------example (sp defined in example above)\n",
    "tt = splitCourses(sp)\n",
    "tt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "cf9354fc",
   "metadata": {},
   "outputs": [],
   "source": [
    "#input: course/department dictionary, and checks length of values in it to see if there are any classes left\n",
    "def checkLengOfVals(course_dict):\n",
    "    ph = list(course_dict.values())    #get a list of all the values (list of numpy arrays)\n",
    "    #print('\\n\\nOUTSIDE FUNC PRINTING TEST VALS')\n",
    "    #print(len(ph))\n",
    "    #print(ph)\n",
    "    #print('\\n\\n')\n",
    "    start = []\n",
    "    for i in ph:                      #use for loop to extarct array vals and create one big arr of strings instead of list of arrays\n",
    "        start = np.append(start, i)\n",
    "    if len(start) > 0:              #check if there are any values in the array return accordingly\n",
    "        return True\n",
    "    else:\n",
    "        return False\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "215439e5",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'ARCHI': array(['ARCHI 120', 'ARCHI 121', 'ARCHI 130', 'ARCHI 131', 'ARCHI 220'],\n",
       "       dtype='<U9'),\n",
       " 'ENGL': array(['ENGL 126'], dtype='<U8'),\n",
       " 'MATH': array(['MATH 192'], dtype='<U8'),\n",
       " 'PHYS': array(['PHYS 130'], dtype='<U8')}"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def removeCourseFromDict(course_dict, course):\n",
    "    abv = course.split()[0]\n",
    "    clist = course_dict[abv]\n",
    "    for i, c in enumerate(clist):\n",
    "        if c == course:\n",
    "            index = i\n",
    "    course_dict[abv] = np.delete(course_dict[abv], index)\n",
    "    return course_dict\n",
    "#-------------------------------------------example (tt defined two cells above)\n",
    "ttv2 = removeCourseFromDict(tt, 'ENGL 122')\n",
    "ttv2\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "3a2e4432",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['PSYCH 101', 'PSYCH 214']\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def canClassBeAdded(classAbv, semesters, i):\n",
    "    if not hasPrereq(classAbv):          #if theres no prereqs true will be returned right away\n",
    "        return True\n",
    "    if i == 0:                          #if you are in the 1st sem and there are prereqs you can't add the class (false returned)\n",
    "        return False\n",
    "    \n",
    "    pres = getPrereqs(classAbv)\n",
    "    if len(pres) == 2:                  #2 is max leng of prereqs. So each class only has 1 or 2 prereqs, but can also have coreqs\n",
    "        if i == 1:\n",
    "            if pres[0] in semesters[i-1].astype(str) and pres[1] in semesters[i-1].astype(str):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "        elif i == 2:\n",
    "            if (pres[0] in semesters[i-2].astype(str) or pres[0] in semesters[i-1].astype(str)) and (pres[1] in semesters[i-2].astype(str) or pres[1] in semesters[i-1].astype(str)):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "        elif i == 3:\n",
    "            if (pres[0] in semesters[i-3].astype(str) or pres[0] in semesters[i-2].astype(str) or pres[0] in semesters[i-1].astype(str)) and (pres[1] in semesters[i-3].astype(str) or pres[1] in semesters[i-2].astype(str) or pres[1] in semesters[i-1].astype(str)):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "    else:                                 #the else case is for when there is only one prereq\n",
    "        if i == 1:\n",
    "            if pres[0] in semesters[i-1].astype(str):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "        elif i == 2:\n",
    "            if pres[0] in semesters[i-2].astype(str) or pres[0] in semesters[i-1].astype(str):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "        elif i == 3:\n",
    "            if pres[0] in semesters[i-3].astype(str) or pres[0] in semesters[i-2].astype(str) or pres[0] in semesters[i-1].astype(str):\n",
    "                return True\n",
    "            else:\n",
    "                return False\n",
    "        \n",
    "    return True\n",
    "\n",
    "#---------------------------------------------------------example\n",
    "\n",
    "testsems = [np.array(['PSYCH 101', 'ENGL 122', 'MATH 142']), np.array(['PHYS 130','PSYCH 214', 'ARCHI 121']), np.array(['ENGL 126', 'ARCHI 130']), np.array([])]\n",
    "i = 3\n",
    "cb = 'PSYCH 215'\n",
    "print(getPrereqs(cb))\n",
    "testn = canClassBeAdded(cb, testsems, i)\n",
    "testn"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "id": "7b1df509",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Required Courses:\n",
      "['CHEM 120' 'CHEM 121' 'COMSC 165' 'ENGL 122' 'ENGL 126' 'MATH 192'\n",
      " 'MATH 193' 'MATH 194' 'MATH 292' 'MATH 294' 'PHYS 130' 'PHYS 230'\n",
      " 'PHYS 231']\n",
      "\n",
      "\n",
      "\n",
      "COURSES PER SEM:  4\n",
      "{'CHEM': array(['CHEM 108', 'CHEM 120', 'CHEM 121'], dtype='<U8'), 'COMSC': array(['COMSC 110', 'COMSC 165'], dtype='<U9'), 'ENGL': array(['ENGL 122', 'ENGL 126'], dtype='<U8'), 'MATH': array(['MATH 192', 'MATH 193', 'MATH 194', 'MATH 292', 'MATH 294'],\n",
      "      dtype='<U8'), 'PHYS': array(['PHYS 130', 'PHYS 230', 'PHYS 231'], dtype='<U8')}\n",
      "\n",
      "\n",
      "Plan:\n",
      "Sem 1\n",
      "['CHEM 108' 'COMSC 110' 'ENGL 122' 'MATH 192']\n",
      "\n",
      "Sem 2\n",
      "['MATH 193' 'PHYS 130' 'CHEM 120' 'COMSC 165']\n",
      "\n",
      "Sem 3\n",
      "['ENGL 126' 'MATH 194' 'MATH 292' 'PHYS 230']\n",
      "\n",
      "Sem 4\n",
      "['CHEM 121' 'MATH 294' 'PHYS 231']\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\user1\\AppData\\Local\\Temp\\ipykernel_15228\\2414598076.py:80: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison\n",
      "  if cos not in semesters[i] and cos not in semesters[i-1]:                         #if  coreq not in plan add it\n",
      "C:\\Users\\user1\\AppData\\Local\\Temp\\ipykernel_15228\\2414598076.py:152: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison\n",
      "  if pres not in semesters[i] and pres not in semesters[i-1] and pres not in semesters[i-2] and pres not in semesters[i-3]:    #make sure prereq is not already in the course plan in this semester or the previous\n"
     ]
    }
   ],
   "source": [
    "#input is arr of required courses, output is a 4 semester plan.\n",
    "def createPlan(reqCourses):\n",
    "    sem1, sem2, sem3, sem4 = np.array([]), np.array([]), np.array([]), np.array([])\n",
    "    semesters = [sem1, sem2, sem3, sem4]            #list of numpy array to hold course for each semester\n",
    "    reqCourses = addPrereqs(reqCourses)             #adding any needed prereqs to the required courses list\n",
    "    reqCourses = addCoreqs(reqCourses)              ##adding any needed coreqs to the required courses list\n",
    "    amtOfCourses = len(reqCourses)                  #getting int number of courses ex: 14\n",
    "    avgSemCourses = int(math.ceil(amtOfCourses/4))   #dividing number by 4 and rounding up to whole number to get approx number of courses per semester\n",
    "    print('COURSES PER SEM: ', avgSemCourses)\n",
    "    #allCourses = set(reqCourses)                    #creating a set to store all classes so i can remove dynamically as they are added to plan and also another check to make sure there are no repeats (sets cant have duplicates)\n",
    "    coursesSplit = splitCourses(reqCourses)         #creating a dictionary of arrs splitting up courses by department \n",
    "    departments = list(coursesSplit.keys())         #getting a list of the keys from the dictionary (list of deparments ex ['BUS', 'COMSC', etc])\n",
    "    i, courseCount = 0, 0\n",
    "    #print(coursesSplit)\n",
    "    lastClass = ''\n",
    "    classAddedInNextSem = False\n",
    "    while(checkLengOfVals(coursesSplit)):             #continue while there are more classes left in course deparment dictionary\n",
    "        for d in departments:\n",
    "            if len(coursesSplit[d]) != 0:                      #checking if list in courses dictionary has values in it\n",
    "                \n",
    "                #individual prereq check\n",
    "                if i == 0:                                 #semester one Prereq check\n",
    "                    prereqAdded = False\n",
    "                    if hasPrereq(coursesSplit[d][0]):      #check if the individual course has prereqs\n",
    "                        ph = getPrereqs(coursesSplit[d][0]) #if so we retrieve them\n",
    "                        for pres in ph:                     #go through each prereq\n",
    "                            if pres not in semesters[i]:    #make sure prereq is not already in the course plan\n",
    "                                semesters[i] = np.append(semesters[i], pres) #if not then append it to the course plan\n",
    "                                coursesSplit = removeCourseFromDict(coursesSplit, pres) #after the courses is added we must remove it from course dictionary\n",
    "                                prereqAdded = True                           #tell the program a prereq was aded\n",
    "                                courseCount +=1                             #increase total course count by 1\n",
    "                                if courseCount % avgSemCourses == 0:       #check if its need to go to next semester\n",
    "                                    i+=1\n",
    "                    \n",
    "                    if not prereqAdded:                                                #if not prereq was added do the followiing\n",
    "                        if hasCoreq(coursesSplit[d][0]):                  #!          #check for coreqs\n",
    "                            ch = getCoreqs(coursesSplit[d][0])            #!          #get coreqs\n",
    "                            for cos in ch:                                         \n",
    "                                if cos not in semesters[i]:                         #if  coreq not in plan add it\n",
    "                                    semesters[i] = np.append(semesters[i], cos)\n",
    "                                    courseCount +=1                                 #inc total course count\n",
    "                                    coursesSplit = removeCourseFromDict(coursesSplit, cos)\n",
    "                                    if courseCount % avgSemCourses == 0:        \n",
    "                                        i+=1     \n",
    "                        if coursesSplit[d][0] == lastClass:                                   #adding this to fix a infinite loop glitch that happens if only one department is left in the dictionary and the class trying to be added has a prereq in the same semester it is in so you must move to the next semester\n",
    "                            semesters[i+1] = np.append(semesters[i+1], coursesSplit[d][0])\n",
    "                            classAddedInNextSem = True\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)\n",
    "                            \n",
    "                        if canClassBeAdded(coursesSplit[d][0], semesters, i):           #making sure can be added\n",
    "                            semesters[i] = np.append(semesters[i], coursesSplit[d][0])  #add the current course to the plan\n",
    "                            courseCount +=1                                             #inc total course count\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)             #deleting from course dict after added\n",
    "                            if courseCount % avgSemCourses == 0:                        #check if need to change semesters\n",
    "                                i+=1\n",
    "                        else:\n",
    "                            lastClass = coursesSplit[d][0]\n",
    "                              \n",
    "#----------------------------------------------------------------------------------------------------------------------------                                \n",
    "                elif i == 1:\n",
    "                    if classAddedInNextSem:\n",
    "                        courseCount += 1\n",
    "                        classAddedInNextSem = False\n",
    "                    prereqAdded = False\n",
    "                    if hasPrereq(coursesSplit[d][0]):      #check if the individual course has prereqs\n",
    "                        ph = getPrereqs(coursesSplit[d][0]) #if so we retrieve them\n",
    "                        for pres in ph:                     #go through each prereq\n",
    "                            if pres not in semesters[i] and pres not in semesters[i-1]:    #make sure prereq is not already in the course plan in this semester or the previous\n",
    "                                semesters[i] = np.append(semesters[i], pres) #if not then append it to the course plan\n",
    "                                coursesSplit = removeCourseFromDict(coursesSplit, pres) #after the courses is added we must remove it from course dictionary\n",
    "                                prereqAdded = True                           #tell the program a prereq was aded\n",
    "                                courseCount +=1                             #increase total course count by 1\n",
    "                                if courseCount % avgSemCourses == 0:       #check if its need to go to next semester\n",
    "                                    i+=1\n",
    "                    \n",
    "                    if not prereqAdded:                                                #if not prereq was added do the followiing\n",
    "                        if hasCoreq(coursesSplit[d][0]):                            #check for coreqs\n",
    "                            ch = getCoreqs(coursesSplit[d][0])                      #get coreqs\n",
    "                            for cos in ch:                                         \n",
    "                                if cos not in semesters[i] and cos not in semesters[i-1]:                         #if  coreq not in plan add it\n",
    "                                    semesters[i] = np.append(semesters[i], cos)\n",
    "                                    courseCount +=1                                 #inc total course count\n",
    "                                    coursesSplit = removeCourseFromDict(coursesSplit, cos)\n",
    "                                    if courseCount % avgSemCourses == 0:        \n",
    "                                        i+=1\n",
    "                        if coursesSplit[d][0] == lastClass:                                   #adding this to fix a infinite loop glitch that happens if only one department is left in the dictionary and the class trying to be added has a prereq in the same semester it is in so you must move to the next semester\n",
    "                            semesters[i+1] = np.append(semesters[i+1], coursesSplit[d][0])\n",
    "                            classAddedInNextSem = True\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)\n",
    "                            \n",
    "                        if canClassBeAdded(coursesSplit[d][0], semesters, i):           #making sure can be added\n",
    "                            semesters[i] = np.append(semesters[i], coursesSplit[d][0])  #add the current course to the plan\n",
    "                            courseCount +=1                                             #inc total course count\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)             #deleting from course dict after added\n",
    "                            if courseCount % avgSemCourses == 0:                        #check if need to change semesters\n",
    "                                i+=1\n",
    "                        else:\n",
    "                            lastClass = coursesSplit[d][0]\n",
    "\n",
    "#-------------------------------------------------------------------------------------------------------------------------------                    \n",
    "                elif i == 2:\n",
    "                    if classAddedInNextSem:\n",
    "                        courseCount += 1\n",
    "                        classAddedInNextSem = False\n",
    "                    prereqAdded = False\n",
    "                    if hasPrereq(coursesSplit[d][0]):      #check if the individual course has prereqs\n",
    "                        ph = getPrereqs(coursesSplit[d][0]) #if so we retrieve them\n",
    "                        for pres in ph:                     #go through each prereq\n",
    "#!!!!!!!!!!!THE IF STATEMENTS ARE GETTING FUTURE WARNING ERRORS, IF THEY BREAK CAN BE FIXED BY CONVERTING SEMESETER NP.ARR TO STRRING USING .astype(str)\n",
    "                            if pres not in semesters[i].astype(str) and pres not in semesters[i-1].astype(str) and pres not in semesters[i-2].astype(str):    #make sure prereq is not already in the course plan in this semester or the previous\n",
    "                                semesters[i] = np.append(semesters[i], pres) #if not then append it to the course plan\n",
    "                                coursesSplit = removeCourseFromDict(coursesSplit, pres) #after the courses is added we must remove it from course dictionary\n",
    "                                prereqAdded = True                           #tell the program a prereq was aded\n",
    "                                courseCount +=1                             #increase total course count by 1\n",
    "                                if courseCount % avgSemCourses == 0:       #check if its need to go to next semester\n",
    "                                    i+=1\n",
    "                    \n",
    "                    if not prereqAdded:                                                #if not prereq was added do the followiing\n",
    "                        if hasCoreq(coursesSplit[d][0]):                            #check for coreqs\n",
    "                            ch = getCoreqs(coursesSplit[d][0])                      #get coreqs\n",
    "                            for cos in ch:                                         \n",
    "                                if cos not in semesters[i] and cos not in semesters[i-1] and cos not in semesters[i-2]:                         #if  coreq not in plan add it\n",
    "                                    semesters[i] = np.append(semesters[i], cos)\n",
    "                                    courseCount +=1                                 #inc total course count\n",
    "                                    coursesSplit = removeCourseFromDict(coursesSplit, cos)\n",
    "                                    if courseCount % avgSemCourses == 0:        \n",
    "                                        i+=1 \n",
    "                        if coursesSplit[d][0] == lastClass:                                   #adding this to fix a infinite loop glitch that happens if only one department is left in the dictionary and the class trying to be added has a prereq in the same semester it is in so you must move to the next semester\n",
    "                            semesters[i+1] = np.append(semesters[i+1], coursesSplit[d][0])\n",
    "                            classAddedInNextSem = True\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)\n",
    "                            \n",
    "                        if canClassBeAdded(coursesSplit[d][0], semesters, i):           #making sure can be added\n",
    "                            semesters[i] = np.append(semesters[i], coursesSplit[d][0])  #add the current course to the plan\n",
    "                            courseCount +=1                                             #inc total course count\n",
    "                            coursesSplit[d] = np.delete(coursesSplit[d], 0)             #deleting from course dict after added\n",
    "                            if courseCount % avgSemCourses == 0:                        #check if need to change semesters\n",
    "                                i+=1\n",
    "                        else:\n",
    "                            lastClass = coursesSplit[d][0]\n",
    "                        \n",
    "                          \n",
    " #------------------------------------------------------------------------------------------------------------------------------\n",
    "                elif i == 3:\n",
    "                    if classAddedInNextSem:\n",
    "                        courseCount += 1\n",
    "                        classAddedInNextSem = False\n",
    "                    prereqAdded = False\n",
    "                    if hasPrereq(coursesSplit[d][0]):      #check if the individual course has prereqs\n",
    "                        ph = getPrereqs(coursesSplit[d][0]) #if so we retrieve them\n",
    "                        for pres in ph:                     #go through each prereq\n",
    "                            if pres not in semesters[i] and pres not in semesters[i-1] and pres not in semesters[i-2] and pres not in semesters[i-3]:    #make sure prereq is not already in the course plan in this semester or the previous\n",
    "                                semesters[i] = np.append(semesters[i], pres) #if not then append it to the course plan\n",
    "                                coursesSplit = removeCourseFromDict(coursesSplit, pres) #after the courses is added we must remove it from course dictionary\n",
    "                                prereqAdded = True                           #tell the program a prereq was aded\n",
    "                                courseCount +=1                             #increase total course count by 1\n",
    "                                if courseCount % avgSemCourses == 0:       #check if its need to go to next semester\n",
    "                                    i+=1\n",
    "                    if not prereqAdded:                                                #if not prereq was added do the followiing\n",
    "                        if hasCoreq(coursesSplit[d][0]):                            #check for coreqs\n",
    "                            ch = getCoreqs(coursesSplit[d][0])                      #get coreqs\n",
    "                            for cos in ch:\n",
    "                                if cos not in semesters[i] and cos not in semesters[i-1] and cos not in semesters[i-2] and cos not in semesters[i-3]:                         #if  coreq not in plan add it\n",
    "                                    semesters[i] = np.append(semesters[i], cos)\n",
    "                                    courseCount +=1                                 #inc total course count\n",
    "                                    coursesSplit = removeCourseFromDict(coursesSplit, cos)\n",
    "                                    if courseCount % avgSemCourses == 0:        \n",
    "                                        i+=1\n",
    "                        \n",
    "                        semesters[i] = np.append(semesters[i], coursesSplit[d][0])  #add the current course to the plan\n",
    "                        courseCount +=1                                             #inc total course count\n",
    "                        coursesSplit[d] = np.delete(coursesSplit[d], 0)             #deleting from course dict after added\n",
    "                        if courseCount % avgSemCourses == 0:                        #check if need to change semesters\n",
    "                            i+=1\n",
    "                \n",
    "                \n",
    "                #semesters[i] = np.append(semesters[i], coursesSplit[d][0])     #retrieve the first value(course) and add it to the sem plan\n",
    "                #courseCount +=1\n",
    "                #coursesSplit[d] = np.delete(coursesSplit[d], 0)\n",
    "                #if courseCount % avgSemCourses == 0:\n",
    "                    #i+=1\n",
    "                \n",
    "            \n",
    "\n",
    "    return semesters\n",
    "   \n",
    "#-------------------------------------------------------------- example\n",
    "co = [ucla]\n",
    "ma = ['Aerospace Engineering']\n",
    "stR = getStudentRequiredCourses(co, ma)\n",
    "print('Required Courses:')\n",
    "print(stR)\n",
    "print('\\n\\n')\n",
    "n = createPlan(stR)\n",
    "print('\\n\\nPlan:')\n",
    "print('Sem 1')\n",
    "print(n[0])\n",
    "print('\\nSem 2')\n",
    "print(n[1])\n",
    "print('\\nSem 3')\n",
    "print(n[2])\n",
    "print('\\nSem 4')\n",
    "print(n[3])\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "555570b0",
   "metadata": {},
   "source": [
    "#### FutureWarning Error Above does not cause any issues but if needed it can be easily fixed by converting the NumPy semester arrays used in the if statements to strings using .astype(str)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "0120640b",
   "metadata": {},
   "source": [
    "# Examples Below"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d65524c9",
   "metadata": {},
   "source": [
    "## Ex of getting Required Courses for two colleges and two majors (majors in the m list correspond to the college in same index in the c list)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "5da5e50b",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array(['CHEM 120', 'CHEM 121', 'COMSC 165', 'ENGL 122', 'ENGL 126',\n",
       "       'MATH 192', 'MATH 193', 'MATH 194', 'MATH 292', 'MATH 294',\n",
       "       'PHYS 130', 'PHYS 230', 'PHYS 231'], dtype='<U9')"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "c = [berkeley, ucla]\n",
    "m = ['Aerospace Engineering', 'Aerospace Engineering']\n",
    "sR = getStudentRequiredCourses(c, m)\n",
    "sR"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "081ea142",
   "metadata": {},
   "source": [
    "## Ex of split by department (uses data from above example)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "0349a012",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'CHEM': array(['CHEM 120', 'CHEM 121'], dtype='<U8'),\n",
       " 'COMSC': array(['COMSC 165'], dtype='<U9'),\n",
       " 'ENGL': array(['ENGL 122', 'ENGL 126'], dtype='<U8'),\n",
       " 'MATH': array(['MATH 192', 'MATH 193', 'MATH 194', 'MATH 292', 'MATH 294'],\n",
       "       dtype='<U8'),\n",
       " 'PHYS': array(['PHYS 130', 'PHYS 230', 'PHYS 231'], dtype='<U8')}"
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cD = splitCourses(sR)\n",
    "cD\n",
    "#a = list(cD.values())\n",
    "#starter = []\n",
    "#for i in a:\n",
    "#    starter = np.append(starter, i)\n",
    "#starter"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "ff1bb81b",
   "metadata": {},
   "source": [
    "## Ex of getting data from the dataframe on a specific class"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "05635c77",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Title     Biological Anthropology\n",
       "Units                           3\n",
       "Prereq                          N\n",
       "Coreq                           N\n",
       "Name: ANTHR 140, dtype: object"
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#n1 = input('Major')\n",
    "#berkeley[n1]['reqCourses'][1]\n",
    "n2 = berkeley['Anthropology']['reqCourses'][0] #gets required class list (list of strs) for anthro transfer to berk and gets 1st class from the list\n",
    "m3 = df.loc[n2] #looking up class in dataframe using string name of class \n",
    "m3"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "28b37a55",
   "metadata": {},
   "source": [
    "## Tail of dataframe"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "4514511e",
   "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>Title</th>\n",
       "      <th>Units</th>\n",
       "      <th>Prereq</th>\n",
       "      <th>Coreq</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Class</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>PHYS 125</th>\n",
       "      <td>Calculus Supplement for Physics 121</td>\n",
       "      <td>0.5</td>\n",
       "      <td>N</td>\n",
       "      <td>PHYS 121-MATH 193</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PHYS 129</th>\n",
       "      <td>Introductory Physics for Engineers</td>\n",
       "      <td>4</td>\n",
       "      <td>N</td>\n",
       "      <td>MATH 192</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PHYS 130</th>\n",
       "      <td>Physics for Engineers and Scientists A: Mechan...</td>\n",
       "      <td>4</td>\n",
       "      <td>N</td>\n",
       "      <td>MATH 193</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PHYS 230</th>\n",
       "      <td>Physics for Engineers and Scientists B: Heat a...</td>\n",
       "      <td>4</td>\n",
       "      <td>PHYS 130</td>\n",
       "      <td>MATH 292</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>PHYS 231</th>\n",
       "      <td>Physics for Engineers and Scientists C: Optics...</td>\n",
       "      <td>4</td>\n",
       "      <td>PHYS 230</td>\n",
       "      <td>MATH 294</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>SPAN 221</th>\n",
       "      <td>Fourth Term Spanish</td>\n",
       "      <td>5</td>\n",
       "      <td>SPAN 220</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>SPAN 230</th>\n",
       "      <td>Fifth Term Spanish</td>\n",
       "      <td>3</td>\n",
       "      <td>SPAN 221</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>SPAN 231</th>\n",
       "      <td>Sixth Term Spanish</td>\n",
       "      <td>3</td>\n",
       "      <td>SPAN 230</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>SPAN 240</th>\n",
       "      <td>Spanish for Heritage Speakers I</td>\n",
       "      <td>5</td>\n",
       "      <td>N</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>SPAN 241</th>\n",
       "      <td>Spanish for Heritage Speakers II</td>\n",
       "      <td>5</td>\n",
       "      <td>SPAN 240</td>\n",
       "      <td>N</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>79 rows × 4 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                      Title Units    Prereq  \\\n",
       "Class                                                                         \n",
       "PHYS 125                Calculus Supplement for Physics 121   0.5         N   \n",
       "PHYS 129                 Introductory Physics for Engineers     4         N   \n",
       "PHYS 130  Physics for Engineers and Scientists A: Mechan...     4         N   \n",
       "PHYS 230  Physics for Engineers and Scientists B: Heat a...     4  PHYS 130   \n",
       "PHYS 231  Physics for Engineers and Scientists C: Optics...     4  PHYS 230   \n",
       "...                                                     ...   ...       ...   \n",
       "SPAN 221                                Fourth Term Spanish     5  SPAN 220   \n",
       "SPAN 230                                 Fifth Term Spanish     3  SPAN 221   \n",
       "SPAN 231                                 Sixth Term Spanish     3  SPAN 230   \n",
       "SPAN 240                    Spanish for Heritage Speakers I     5         N   \n",
       "SPAN 241                   Spanish for Heritage Speakers II     5  SPAN 240   \n",
       "\n",
       "                      Coreq  \n",
       "Class                        \n",
       "PHYS 125  PHYS 121-MATH 193  \n",
       "PHYS 129           MATH 192  \n",
       "PHYS 130           MATH 193  \n",
       "PHYS 230           MATH 292  \n",
       "PHYS 231           MATH 294  \n",
       "...                     ...  \n",
       "SPAN 221                  N  \n",
       "SPAN 230                  N  \n",
       "SPAN 231                  N  \n",
       "SPAN 240                  N  \n",
       "SPAN 241                  N  \n",
       "\n",
       "[79 rows x 4 columns]"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.tail(79)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}