Project #1 - Predicting NFL Draft Team Needs-Copy2.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# OVERVIEW\n",
    "# Data\n",
    "# What position each team drafted and what their stats for the previous season was\n",
    "# Chargers 2020 Draft QB, LB - Chargers 2019 receiving, rushing, tackles, ints, points for, etc\n",
    "    \n",
    "# 1st Round Draft History Data - http://www.drafthistory.com/index.php/rounds/round_1\n",
    "# Team Stats and Rankings - https://www.pro-football-reference.com/teams/cin/2019.htm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# IMPORTS\n",
    "# Data\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "\n",
    "# Visualization\n",
    "import seaborn as sns\n",
    "import matplotlib.pyplot as plt\n",
    "\n",
    "# ML Models\n",
    "from sklearn.linear_model import LogisticRegression\n",
    "from sklearn.svm import SVC, LinearSVC\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.neighbors import KNeighborsClassifier\n",
    "from sklearn.naive_bayes import GaussianNB\n",
    "from sklearn.linear_model import Perceptron\n",
    "from sklearn.linear_model import SGDClassifier\n",
    "from sklearn.tree import DecisionTreeClassifier\n",
    "\n",
    "# Web Scraping\n",
    "from requests import get\n",
    "from bs4 import BeautifulSoup\n",
    "\n",
    "# Math\n",
    "import math"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "32\n"
     ]
    }
   ],
   "source": [
    "# CONSTANTS\n",
    "CURRENT_YEAR = 2020\n",
    "YEARS = 20\n",
    "TEAM_DICT = {'Bengals':'cin','Redskins':'was','Lions':'det','Giants':'nyg','Dolphins':'mia','Chargers':'sdg',\n",
    "            'Panthers':'car','Cardinals':'crd','Jaguars':'jax','Browns':'cle','Jets':'nyj','Raiders':'rai',\n",
    "             'Buccaneers':'tam','49ers':'sfo','Broncos':'den','Falcons':'atl','Cowboys':'dal','Eagles':'phi',\n",
    "             'Vikings':'min','Saints':'nor','Packers':'gnb','Seahawks':'sea','Ravens':'rav','Titans':'oti',\n",
    "             'Chiefs':'kan','Colts':'clt','Texans':'htx','Bears':'chi','Steelers':'pit','Rams':'ram','Bills':'buf',\n",
    "             'Patriots':'nwe'}\n",
    "# POS_DICT = {'QB':1,'RB':2,'WR':3,'TE':4,' C':5,' G':6,' T':7,'DT':8,'DE':9,'LB':10,'DB':11}\n",
    "EMPTY_COLS = ['Off. Rank Ply', 'Off. Rank Y/P', 'Off. Rank Cmp', 'Off. Rank 1stD', 'Off. Rank 1stD', 'Off. Rank Pen',\n",
    "             'Off. Rank Yds', 'Off. Rank 1stPy', 'Off. Rank #Dr', 'Def. Rank Ply', 'Def. Rank Y/P', 'Def. Rank Cmp',\n",
    "             'Def. Rank 1stD', 'Def. Rank 1stD', 'Def. Rank Pen', 'Def. Rank Yds', 'Def. Rank 1stPy', 'Def. Rank #Dr',\n",
    "             'Off. Rank 1stD.1', 'Off. Rank 1stD.2', 'Off. Rank Yds.3', 'Def. Rank 1stD.1', 'Def. Rank 1stD.2', 'Def. Rank Yds.3']\n",
    "UNIT_DICT = {'QB':'OFF','RB':'OFF','WR':'OFF','TE':'OFF',' C':'OFF',' G':'OFF',' T':'OFF','DT':'DEF','DE':'DEF','LB':'DEF',\n",
    "             'DB':'DEF'}\n",
    "NUM_NEIGHBORS = 5\n",
    "print(len(TEAM_DICT))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# WEB SCRAPING FUNCTIONS\n",
    "def get_first_rounders() -> list:\n",
    "    url = 'http://www.drafthistory.com/index.php/rounds/round_1'\n",
    "    response = get(url)\n",
    "    soup = BeautifulSoup(response.text, 'html.parser')\n",
    "    table = soup.find(id='main')\n",
    "    row_containers = table.table.find_all('tr')\n",
    "    row_containers = row_containers[2:]https://www.youtube.com/playlist?list=WL\n",
    "    labels = ['Year','Round','Pick','Player','Name','Team','Position','College']\n",
    "    \n",
    "    all_rows = []\n",
    "    should_break = False\n",
    "    for i in row_containers:\n",
    "        new_row = []\n",
    "        cols = i.find_all('td')\n",
    "        for j in range(0, len(cols)):\n",
    "            if j == 0:\n",
    "                try:\n",
    "                    year = int(cols[j].text)\n",
    "                    if year == CURRENT_YEAR - YEARS:\n",
    "                        should_break=True\n",
    "                        break\n",
    "                    new_row.append(cols[j].text)\n",
    "                except:\n",
    "                    new_row.append(year)\n",
    "            else:\n",
    "                new_row.append(cols[j].text)\n",
    "        if should_break:\n",
    "            break\n",
    "        if new_row:\n",
    "            all_rows.append(new_row)\n",
    "    \n",
    "    df = pd.DataFrame(all_rows,columns=labels)\n",
    "    df.drop(columns=['Round','Player','College'], inplace=True)\n",
    "    return df\n",
    "\n",
    "def get_team_stats_by_year_url(team_name: str, year) -> str:\n",
    "    base_url = 'https://www.pro-football-reference.com/teams/'\n",
    "    end_url = '.htm'\n",
    "    return base_url + TEAM_DICT[team_name] + '/' + str(year) + end_url\n",
    "\n",
    "def get_team_stats(url) -> 'DataFrame':\n",
    "    response = get(url)\n",
    "    if response.status_code == 404:\n",
    "        return pd.DataFrame()\n",
    "    soup = BeautifulSoup(response.text,'html.parser')\n",
    "    table_container = soup.find(id='all_team_stats')\n",
    "    if table_container:\n",
    "        print('Scraping', url)\n",
    "    else:\n",
    "        print(url)\n",
    "    table = table_container.find(class_='table_outer_container').div.table\n",
    "    \n",
    "    label_containers = table.thead.find_all('tr')[1].find_all('th')\n",
    "    labels = []\n",
    "    for i in label_containers:\n",
    "        labels.append(str(i.text))\n",
    "        \n",
    "    list_of_rows = []\n",
    "    row_containers = table.tbody.find_all('tr')\n",
    "    for i in row_containers:\n",
    "        row = []\n",
    "        row.append(i.th)\n",
    "        stat_containers = i.find_all('td')\n",
    "        for j in stat_containers:\n",
    "            try: \n",
    "                num = float(j.text)\n",
    "                row.append(num)\n",
    "            except:\n",
    "                row.append(str(j.text).strip())\n",
    "        list_of_rows.append(row)\n",
    "    \n",
    "    return pd.DataFrame(list_of_rows,columns=labels)\n",
    "\n",
    "def convert_team_stats_to_one_row(df) -> 'DataFrame':\n",
    "    team_columns = []\n",
    "    opp_columns = []\n",
    "    off_rank_columns = []\n",
    "    def_rank_columns = []\n",
    "    \n",
    "    for i in df.columns:\n",
    "        team_columns.append('Team ' + i)\n",
    "        opp_columns.append('Opp. ' + i)\n",
    "        off_rank_columns.append('Off. Rank ' + i)\n",
    "        def_rank_columns.append('Def. Rank ' + i)\n",
    "        \n",
    "    row1 = pd.DataFrame([list(df.loc[0])], columns=team_columns)\n",
    "    row2 = pd.DataFrame([list(df.loc[1])], columns=opp_columns)\n",
    "    row3 = pd.DataFrame([list(df.loc[2])], columns=off_rank_columns)\n",
    "    row4 = pd.DataFrame([list(df.loc[3])], columns=def_rank_columns)\n",
    "    \n",
    "    new_df = pd.concat([row1, row2, row3, row4], axis=1).reindex(row1.index)\n",
    "    return new_df\n",
    "\n",
    "def make_final_one_row(df, draft_year, pick_number, team_name, position_drafted) -> 'DataFrame':\n",
    "    df.drop(columns=['Team Player', 'Opp. Player', 'Off. Rank Player', 'Def. Rank Player'], inplace=True)\n",
    "    df.insert(0, 'Pick Number', [pick_number], True)\n",
    "    df.insert(0, 'Draft Year', [draft_year], True)\n",
    "    df.insert(0, 'Team', [team_name], True)\n",
    "    df.insert(0, 'Position Drafted', [position_drafted], True)\n",
    "    return df\n",
    "\n",
    "def get_final_df() -> 'DataFrame':\n",
    "    first_rounders = get_first_rounders()\n",
    "    frames = []\n",
    "    for i in first_rounders.index:\n",
    "        row = list(first_rounders.loc[i])\n",
    "\n",
    "        draft_year = row[0]\n",
    "        pick_number = row[1]\n",
    "        team_name = row[3]\n",
    "        position_drafted = row[4]\n",
    "\n",
    "        stats_url = get_team_stats_by_year_url(team_name, int(draft_year)-1)\n",
    "        team_stats_df = get_team_stats(stats_url)\n",
    "        if team_stats_df.empty:\n",
    "            print('Not Found', stats_url)\n",
    "            continue\n",
    "        else:\n",
    "            pass\n",
    "        one_row = convert_team_stats_to_one_row(team_stats_df)\n",
    "        final_row = make_final_one_row(one_row, draft_year, pick_number, team_name, position_drafted)\n",
    "        frames.append(final_row)\n",
    "    final_df = pd.concat(frames)\n",
    "    return final_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "final_df = get_final_df()\n",
    "final_df.to_csv('Profile For Last ' + str(YEARS) + ' Drafts.csv', encoding = 'utf-8-sig', index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "final_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "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>Position Drafted</th>\n",
       "      <th>Team</th>\n",
       "      <th>Draft Year</th>\n",
       "      <th>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.3</th>\n",
       "      <th>Def. Rank 1stPy</th>\n",
       "      <th>Def. Rank #Dr</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>QB</td>\n",
       "      <td>Bengals</td>\n",
       "      <td>2020</td>\n",
       "      <td>1</td>\n",
       "      <td>279.0</td>\n",
       "      <td>5169.0</td>\n",
       "      <td>1049.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>9.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>DE</td>\n",
       "      <td>Redskins</td>\n",
       "      <td>2020</td>\n",
       "      <td>2</td>\n",
       "      <td>266.0</td>\n",
       "      <td>4395.0</td>\n",
       "      <td>885.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>4.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>DB</td>\n",
       "      <td>Lions</td>\n",
       "      <td>2020</td>\n",
       "      <td>3</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5549.0</td>\n",
       "      <td>1021.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>23.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>7.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>T</td>\n",
       "      <td>Giants</td>\n",
       "      <td>2020</td>\n",
       "      <td>4</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5416.0</td>\n",
       "      <td>1012.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>33.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>8.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>QB</td>\n",
       "      <td>Dolphins</td>\n",
       "      <td>2020</td>\n",
       "      <td>5</td>\n",
       "      <td>306.0</td>\n",
       "      <td>4960.0</td>\n",
       "      <td>1022.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>26.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>631</th>\n",
       "      <td>RB</td>\n",
       "      <td>Vikings</td>\n",
       "      <td>2001</td>\n",
       "      <td>27</td>\n",
       "      <td>397.0</td>\n",
       "      <td>5961.0</td>\n",
       "      <td>958.0</td>\n",
       "      <td>6.2</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>6.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>27.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>632</th>\n",
       "      <td>DB</td>\n",
       "      <td>Raiders</td>\n",
       "      <td>2001</td>\n",
       "      <td>28</td>\n",
       "      <td>479.0</td>\n",
       "      <td>5776.0</td>\n",
       "      <td>1023.0</td>\n",
       "      <td>5.6</td>\n",
       "      <td>20.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>24.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>633</th>\n",
       "      <td>DE</td>\n",
       "      <td>Rams</td>\n",
       "      <td>2001</td>\n",
       "      <td>29</td>\n",
       "      <td>540.0</td>\n",
       "      <td>7075.0</td>\n",
       "      <td>1014.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>634</th>\n",
       "      <td>WR</td>\n",
       "      <td>Colts</td>\n",
       "      <td>2001</td>\n",
       "      <td>30</td>\n",
       "      <td>429.0</td>\n",
       "      <td>6141.0</td>\n",
       "      <td>1026.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>12.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>635</th>\n",
       "      <td>TE</td>\n",
       "      <td>Ravens</td>\n",
       "      <td>2001</td>\n",
       "      <td>31</td>\n",
       "      <td>333.0</td>\n",
       "      <td>5014.0</td>\n",
       "      <td>1058.0</td>\n",
       "      <td>4.7</td>\n",
       "      <td>26.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>31.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>636 rows × 124 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    Position Drafted      Team  Draft Year  Pick Number  Team PF  Team Yds  \\\n",
       "0                 QB   Bengals        2020            1    279.0    5169.0   \n",
       "1                 DE  Redskins        2020            2    266.0    4395.0   \n",
       "2                 DB     Lions        2020            3    341.0    5549.0   \n",
       "3                  T    Giants        2020            4    341.0    5416.0   \n",
       "4                 QB  Dolphins        2020            5    306.0    4960.0   \n",
       "..               ...       ...         ...          ...      ...       ...   \n",
       "631               RB   Vikings        2001           27    397.0    5961.0   \n",
       "632               DB   Raiders        2001           28    479.0    5776.0   \n",
       "633               DE      Rams        2001           29    540.0    7075.0   \n",
       "634               WR     Colts        2001           30    429.0    6141.0   \n",
       "635               TE    Ravens        2001           31    333.0    5014.0   \n",
       "\n",
       "     Team Ply  Team Y/P  Team TO  Team FL  ...  Def. Rank Yds.3  \\\n",
       "0      1049.0       4.9     30.0     14.0  ...              NaN   \n",
       "1       885.0       5.0     21.0      8.0  ...              NaN   \n",
       "2      1021.0       5.4     23.0      8.0  ...              NaN   \n",
       "3      1012.0       5.4     33.0     16.0  ...              NaN   \n",
       "4      1022.0       4.9     26.0      8.0  ...              NaN   \n",
       "..        ...       ...      ...      ...  ...              ...   \n",
       "631     958.0       6.2     28.0     10.0  ...              NaN   \n",
       "632    1023.0       5.6     20.0      9.0  ...              NaN   \n",
       "633    1014.0       7.0     35.0     12.0  ...              NaN   \n",
       "634    1026.0       6.0     29.0     14.0  ...              NaN   \n",
       "635    1058.0       4.7     26.0      7.0  ...              NaN   \n",
       "\n",
       "     Def. Rank 1stPy  Def. Rank #Dr  Def. Rank Sc%  Def. Rank TO%  \\\n",
       "0                NaN            NaN            9.0           27.0   \n",
       "1                NaN            NaN            4.0           15.0   \n",
       "2                NaN            NaN            7.0           25.0   \n",
       "3                NaN            NaN            8.0           29.0   \n",
       "4                NaN            NaN            1.0           28.0   \n",
       "..               ...            ...            ...            ...   \n",
       "631              NaN            NaN            6.0           31.0   \n",
       "632              NaN            NaN           24.0            7.0   \n",
       "633              NaN            NaN            2.0           22.0   \n",
       "634              NaN            NaN           12.0           23.0   \n",
       "635              NaN            NaN           31.0            1.0   \n",
       "\n",
       "     Def. Rank Start  Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  \\\n",
       "0               12.0            18.0             15.0             28.0   \n",
       "1               31.0            31.0             29.0             29.0   \n",
       "2                3.0            21.0             28.0             30.0   \n",
       "3               25.0            19.0             17.0             20.0   \n",
       "4               27.0            20.0             20.0             26.0   \n",
       "..               ...             ...              ...              ...   \n",
       "631              7.0            30.0             30.0             30.0   \n",
       "632              1.0             5.0             15.0             18.0   \n",
       "633             25.0            15.0             14.0             24.0   \n",
       "634              3.0            25.0             28.0             25.0   \n",
       "635              2.0             1.0              1.0              2.0   \n",
       "\n",
       "     Def. Rank Pts  \n",
       "0             22.0  \n",
       "1             30.0  \n",
       "2             24.0  \n",
       "3             25.0  \n",
       "4             32.0  \n",
       "..             ...  \n",
       "631           27.0  \n",
       "632           11.0  \n",
       "633           30.0  \n",
       "634           19.0  \n",
       "635            1.0  \n",
       "\n",
       "[636 rows x 124 columns]"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "final_df = pd.read_csv('Profile For Last ' + str(YEARS) + ' Drafts.csv', encoding = 'utf-8')\n",
    "final_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [],
   "source": [
    "def clean_final_df(final_df) -> 'DataFrame':\n",
    "    \n",
    "    team_start_list = []\n",
    "    opp_start_list = []\n",
    "    for i in list(final_df['Team Start']):\n",
    "        team_start_list.append(float(str(i)[4:].strip()))\n",
    "    for i in list(final_df['Opp. Start']):\n",
    "        opp_start_list.append(float(str(i)[4:].strip()))\n",
    "    \n",
    "    final_df['Team Start'] = team_start_list\n",
    "    final_df['Opp. Start'] = opp_start_list\n",
    "    \n",
    "    team_time_list = []\n",
    "    opp_time_list = []\n",
    "    for i in list(final_df['Team Time']):\n",
    "        team_time_list.append(float((int(str(i)[0])*60)+int(str(i)[2:])))\n",
    "    for i in list(final_df['Opp. Time']):\n",
    "        opp_time_list.append(float((int(str(i)[0])*60)+int(str(i)[2:])))\n",
    "        \n",
    "    final_df['Team Time'] = team_time_list\n",
    "    final_df['Opp. Time'] = opp_time_list\n",
    "    \n",
    "    for i in final_df:\n",
    "        if i in EMPTY_COLS:\n",
    "            final_df = final_df.drop(columns=i)\n",
    "    \n",
    "    return final_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "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>Position Drafted</th>\n",
       "      <th>Team</th>\n",
       "      <th>Draft Year</th>\n",
       "      <th>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>QB</td>\n",
       "      <td>Bengals</td>\n",
       "      <td>2020</td>\n",
       "      <td>1</td>\n",
       "      <td>279.0</td>\n",
       "      <td>5169.0</td>\n",
       "      <td>1049.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>DE</td>\n",
       "      <td>Redskins</td>\n",
       "      <td>2020</td>\n",
       "      <td>2</td>\n",
       "      <td>266.0</td>\n",
       "      <td>4395.0</td>\n",
       "      <td>885.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>31.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>DB</td>\n",
       "      <td>Lions</td>\n",
       "      <td>2020</td>\n",
       "      <td>3</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5549.0</td>\n",
       "      <td>1021.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>23.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>21.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>T</td>\n",
       "      <td>Giants</td>\n",
       "      <td>2020</td>\n",
       "      <td>4</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5416.0</td>\n",
       "      <td>1012.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>33.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>...</td>\n",
       "      <td>20.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>QB</td>\n",
       "      <td>Dolphins</td>\n",
       "      <td>2020</td>\n",
       "      <td>5</td>\n",
       "      <td>306.0</td>\n",
       "      <td>4960.0</td>\n",
       "      <td>1022.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>26.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>27.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>631</th>\n",
       "      <td>RB</td>\n",
       "      <td>Vikings</td>\n",
       "      <td>2001</td>\n",
       "      <td>27</td>\n",
       "      <td>397.0</td>\n",
       "      <td>5961.0</td>\n",
       "      <td>958.0</td>\n",
       "      <td>6.2</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>27.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>632</th>\n",
       "      <td>DB</td>\n",
       "      <td>Raiders</td>\n",
       "      <td>2001</td>\n",
       "      <td>28</td>\n",
       "      <td>479.0</td>\n",
       "      <td>5776.0</td>\n",
       "      <td>1023.0</td>\n",
       "      <td>5.6</td>\n",
       "      <td>20.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>...</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>633</th>\n",
       "      <td>DE</td>\n",
       "      <td>Rams</td>\n",
       "      <td>2001</td>\n",
       "      <td>29</td>\n",
       "      <td>540.0</td>\n",
       "      <td>7075.0</td>\n",
       "      <td>1014.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>634</th>\n",
       "      <td>WR</td>\n",
       "      <td>Colts</td>\n",
       "      <td>2001</td>\n",
       "      <td>30</td>\n",
       "      <td>429.0</td>\n",
       "      <td>6141.0</td>\n",
       "      <td>1026.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>635</th>\n",
       "      <td>TE</td>\n",
       "      <td>Ravens</td>\n",
       "      <td>2001</td>\n",
       "      <td>31</td>\n",
       "      <td>333.0</td>\n",
       "      <td>5014.0</td>\n",
       "      <td>1058.0</td>\n",
       "      <td>4.7</td>\n",
       "      <td>26.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>...</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>636 rows × 102 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    Position Drafted      Team  Draft Year  Pick Number  Team PF  Team Yds  \\\n",
       "0                 QB   Bengals        2020            1    279.0    5169.0   \n",
       "1                 DE  Redskins        2020            2    266.0    4395.0   \n",
       "2                 DB     Lions        2020            3    341.0    5549.0   \n",
       "3                  T    Giants        2020            4    341.0    5416.0   \n",
       "4                 QB  Dolphins        2020            5    306.0    4960.0   \n",
       "..               ...       ...         ...          ...      ...       ...   \n",
       "631               RB   Vikings        2001           27    397.0    5961.0   \n",
       "632               DB   Raiders        2001           28    479.0    5776.0   \n",
       "633               DE      Rams        2001           29    540.0    7075.0   \n",
       "634               WR     Colts        2001           30    429.0    6141.0   \n",
       "635               TE    Ravens        2001           31    333.0    5014.0   \n",
       "\n",
       "     Team Ply  Team Y/P  Team TO  Team FL  ...  Def. Rank Yds.2  \\\n",
       "0      1049.0       4.9     30.0     14.0  ...             32.0   \n",
       "1       885.0       5.0     21.0      8.0  ...             31.0   \n",
       "2      1021.0       5.4     23.0      8.0  ...             21.0   \n",
       "3      1012.0       5.4     33.0     16.0  ...             20.0   \n",
       "4      1022.0       4.9     26.0      8.0  ...             27.0   \n",
       "..        ...       ...      ...      ...  ...              ...   \n",
       "631     958.0       6.2     28.0     10.0  ...             15.0   \n",
       "632    1023.0       5.6     20.0      9.0  ...              5.0   \n",
       "633    1014.0       7.0     35.0     12.0  ...             13.0   \n",
       "634    1026.0       6.0     29.0     14.0  ...             25.0   \n",
       "635    1058.0       4.7     26.0      7.0  ...              1.0   \n",
       "\n",
       "     Def. Rank TD.1  Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  \\\n",
       "0              27.0           25.0            9.0           27.0   \n",
       "1              17.0           26.0            4.0           15.0   \n",
       "2              14.0           10.0            7.0           25.0   \n",
       "3              28.0            4.0            8.0           29.0   \n",
       "4              21.0           22.0            1.0           28.0   \n",
       "..              ...            ...            ...            ...   \n",
       "631            25.0           27.0            6.0           31.0   \n",
       "632             5.0           14.0           24.0            7.0   \n",
       "633            27.0           25.0            2.0           22.0   \n",
       "634            16.0           23.0           12.0           23.0   \n",
       "635             1.0            1.0           31.0            1.0   \n",
       "\n",
       "     Def. Rank Start  Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  \\\n",
       "0               12.0            18.0             15.0             28.0   \n",
       "1               31.0            31.0             29.0             29.0   \n",
       "2                3.0            21.0             28.0             30.0   \n",
       "3               25.0            19.0             17.0             20.0   \n",
       "4               27.0            20.0             20.0             26.0   \n",
       "..               ...             ...              ...              ...   \n",
       "631              7.0            30.0             30.0             30.0   \n",
       "632              1.0             5.0             15.0             18.0   \n",
       "633             25.0            15.0             14.0             24.0   \n",
       "634              3.0            25.0             28.0             25.0   \n",
       "635              2.0             1.0              1.0              2.0   \n",
       "\n",
       "     Def. Rank Pts  \n",
       "0             22.0  \n",
       "1             30.0  \n",
       "2             24.0  \n",
       "3             25.0  \n",
       "4             32.0  \n",
       "..             ...  \n",
       "631           27.0  \n",
       "632           11.0  \n",
       "633           30.0  \n",
       "634           19.0  \n",
       "635            1.0  \n",
       "\n",
       "[636 rows x 102 columns]"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "clean_final_df = clean_final_df(final_df)\n",
    "clean_final_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "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>Position Drafted</th>\n",
       "      <th>Team</th>\n",
       "      <th>Draft Year</th>\n",
       "      <th>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>QB</td>\n",
       "      <td>Bengals</td>\n",
       "      <td>2020</td>\n",
       "      <td>1</td>\n",
       "      <td>279.0</td>\n",
       "      <td>5169.0</td>\n",
       "      <td>1049.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>DE</td>\n",
       "      <td>Redskins</td>\n",
       "      <td>2020</td>\n",
       "      <td>2</td>\n",
       "      <td>266.0</td>\n",
       "      <td>4395.0</td>\n",
       "      <td>885.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>31.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>DB</td>\n",
       "      <td>Lions</td>\n",
       "      <td>2020</td>\n",
       "      <td>3</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5549.0</td>\n",
       "      <td>1021.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>23.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>21.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>T</td>\n",
       "      <td>Giants</td>\n",
       "      <td>2020</td>\n",
       "      <td>4</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5416.0</td>\n",
       "      <td>1012.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>33.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>...</td>\n",
       "      <td>20.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>QB</td>\n",
       "      <td>Dolphins</td>\n",
       "      <td>2020</td>\n",
       "      <td>5</td>\n",
       "      <td>306.0</td>\n",
       "      <td>4960.0</td>\n",
       "      <td>1022.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>26.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>...</td>\n",
       "      <td>27.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>631</th>\n",
       "      <td>RB</td>\n",
       "      <td>Vikings</td>\n",
       "      <td>2001</td>\n",
       "      <td>27</td>\n",
       "      <td>397.0</td>\n",
       "      <td>5961.0</td>\n",
       "      <td>958.0</td>\n",
       "      <td>6.2</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>27.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>632</th>\n",
       "      <td>DB</td>\n",
       "      <td>Raiders</td>\n",
       "      <td>2001</td>\n",
       "      <td>28</td>\n",
       "      <td>479.0</td>\n",
       "      <td>5776.0</td>\n",
       "      <td>1023.0</td>\n",
       "      <td>5.6</td>\n",
       "      <td>20.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>...</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>633</th>\n",
       "      <td>DE</td>\n",
       "      <td>Rams</td>\n",
       "      <td>2001</td>\n",
       "      <td>29</td>\n",
       "      <td>540.0</td>\n",
       "      <td>7075.0</td>\n",
       "      <td>1014.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>634</th>\n",
       "      <td>WR</td>\n",
       "      <td>Colts</td>\n",
       "      <td>2001</td>\n",
       "      <td>30</td>\n",
       "      <td>429.0</td>\n",
       "      <td>6141.0</td>\n",
       "      <td>1026.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>...</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>635</th>\n",
       "      <td>TE</td>\n",
       "      <td>Ravens</td>\n",
       "      <td>2001</td>\n",
       "      <td>31</td>\n",
       "      <td>333.0</td>\n",
       "      <td>5014.0</td>\n",
       "      <td>1058.0</td>\n",
       "      <td>4.7</td>\n",
       "      <td>26.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>...</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>636 rows × 102 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    Position Drafted      Team  Draft Year  Pick Number  Team PF  Team Yds  \\\n",
       "0                 QB   Bengals        2020            1    279.0    5169.0   \n",
       "1                 DE  Redskins        2020            2    266.0    4395.0   \n",
       "2                 DB     Lions        2020            3    341.0    5549.0   \n",
       "3                  T    Giants        2020            4    341.0    5416.0   \n",
       "4                 QB  Dolphins        2020            5    306.0    4960.0   \n",
       "..               ...       ...         ...          ...      ...       ...   \n",
       "631               RB   Vikings        2001           27    397.0    5961.0   \n",
       "632               DB   Raiders        2001           28    479.0    5776.0   \n",
       "633               DE      Rams        2001           29    540.0    7075.0   \n",
       "634               WR     Colts        2001           30    429.0    6141.0   \n",
       "635               TE    Ravens        2001           31    333.0    5014.0   \n",
       "\n",
       "     Team Ply  Team Y/P  Team TO  Team FL  ...  Def. Rank Yds.2  \\\n",
       "0      1049.0       4.9     30.0     14.0  ...             32.0   \n",
       "1       885.0       5.0     21.0      8.0  ...             31.0   \n",
       "2      1021.0       5.4     23.0      8.0  ...             21.0   \n",
       "3      1012.0       5.4     33.0     16.0  ...             20.0   \n",
       "4      1022.0       4.9     26.0      8.0  ...             27.0   \n",
       "..        ...       ...      ...      ...  ...              ...   \n",
       "631     958.0       6.2     28.0     10.0  ...             15.0   \n",
       "632    1023.0       5.6     20.0      9.0  ...              5.0   \n",
       "633    1014.0       7.0     35.0     12.0  ...             13.0   \n",
       "634    1026.0       6.0     29.0     14.0  ...             25.0   \n",
       "635    1058.0       4.7     26.0      7.0  ...              1.0   \n",
       "\n",
       "     Def. Rank TD.1  Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  \\\n",
       "0              27.0           25.0            9.0           27.0   \n",
       "1              17.0           26.0            4.0           15.0   \n",
       "2              14.0           10.0            7.0           25.0   \n",
       "3              28.0            4.0            8.0           29.0   \n",
       "4              21.0           22.0            1.0           28.0   \n",
       "..              ...            ...            ...            ...   \n",
       "631            25.0           27.0            6.0           31.0   \n",
       "632             5.0           14.0           24.0            7.0   \n",
       "633            27.0           25.0            2.0           22.0   \n",
       "634            16.0           23.0           12.0           23.0   \n",
       "635             1.0            1.0           31.0            1.0   \n",
       "\n",
       "     Def. Rank Start  Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  \\\n",
       "0               12.0            18.0             15.0             28.0   \n",
       "1               31.0            31.0             29.0             29.0   \n",
       "2                3.0            21.0             28.0             30.0   \n",
       "3               25.0            19.0             17.0             20.0   \n",
       "4               27.0            20.0             20.0             26.0   \n",
       "..               ...             ...              ...              ...   \n",
       "631              7.0            30.0             30.0             30.0   \n",
       "632              1.0             5.0             15.0             18.0   \n",
       "633             25.0            15.0             14.0             24.0   \n",
       "634              3.0            25.0             28.0             25.0   \n",
       "635              2.0             1.0              1.0              2.0   \n",
       "\n",
       "     Def. Rank Pts  \n",
       "0             22.0  \n",
       "1             30.0  \n",
       "2             24.0  \n",
       "3             25.0  \n",
       "4             32.0  \n",
       "..             ...  \n",
       "631           27.0  \n",
       "632           11.0  \n",
       "633           30.0  \n",
       "634           19.0  \n",
       "635            1.0  \n",
       "\n",
       "[636 rows x 102 columns]"
      ]
     },
     "execution_count": 31,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "specialized_df = clean_final_df\n",
    "reference_df = specialized_df.copy(deep=True)\n",
    "reference_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "def convert_final_df_to_num(final_df) -> 'DataFrame':\n",
    "    years_list = []\n",
    "    for i in list(final_df['Draft Year']):\n",
    "        years_list.append(int(i))\n",
    "    final_df.index = years_list\n",
    "    \n",
    "    num_df = final_df.drop(columns=['Position Drafted','Team','Draft Year'])\n",
    "    # num_df['Position Drafted'] = num_df['Position Drafted'].map(POS_DICT)\n",
    "    return num_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "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>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>Team 1stD</th>\n",
       "      <th>Team Cmp</th>\n",
       "      <th>Team Att</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
       "      <td>1</td>\n",
       "      <td>279.0</td>\n",
       "      <td>5169.0</td>\n",
       "      <td>1049.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>312.0</td>\n",
       "      <td>356.0</td>\n",
       "      <td>616.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
       "      <td>2</td>\n",
       "      <td>266.0</td>\n",
       "      <td>4395.0</td>\n",
       "      <td>885.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>248.0</td>\n",
       "      <td>298.0</td>\n",
       "      <td>479.0</td>\n",
       "      <td>...</td>\n",
       "      <td>31.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
       "      <td>3</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5549.0</td>\n",
       "      <td>1021.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>23.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>313.0</td>\n",
       "      <td>344.0</td>\n",
       "      <td>571.0</td>\n",
       "      <td>...</td>\n",
       "      <td>21.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
       "      <td>4</td>\n",
       "      <td>341.0</td>\n",
       "      <td>5416.0</td>\n",
       "      <td>1012.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>33.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>311.0</td>\n",
       "      <td>376.0</td>\n",
       "      <td>607.0</td>\n",
       "      <td>...</td>\n",
       "      <td>20.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
       "      <td>5</td>\n",
       "      <td>306.0</td>\n",
       "      <td>4960.0</td>\n",
       "      <td>1022.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>26.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>315.0</td>\n",
       "      <td>371.0</td>\n",
       "      <td>615.0</td>\n",
       "      <td>...</td>\n",
       "      <td>27.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
       "      <td>27</td>\n",
       "      <td>397.0</td>\n",
       "      <td>5961.0</td>\n",
       "      <td>958.0</td>\n",
       "      <td>6.2</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>319.0</td>\n",
       "      <td>307.0</td>\n",
       "      <td>495.0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>27.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
       "      <td>28</td>\n",
       "      <td>479.0</td>\n",
       "      <td>5776.0</td>\n",
       "      <td>1023.0</td>\n",
       "      <td>5.6</td>\n",
       "      <td>20.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>337.0</td>\n",
       "      <td>284.0</td>\n",
       "      <td>475.0</td>\n",
       "      <td>...</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
       "      <td>29</td>\n",
       "      <td>540.0</td>\n",
       "      <td>7075.0</td>\n",
       "      <td>1014.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>380.0</td>\n",
       "      <td>380.0</td>\n",
       "      <td>587.0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
       "      <td>30</td>\n",
       "      <td>429.0</td>\n",
       "      <td>6141.0</td>\n",
       "      <td>1026.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>357.0</td>\n",
       "      <td>357.0</td>\n",
       "      <td>571.0</td>\n",
       "      <td>...</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
       "      <td>31</td>\n",
       "      <td>333.0</td>\n",
       "      <td>5014.0</td>\n",
       "      <td>1058.0</td>\n",
       "      <td>4.7</td>\n",
       "      <td>26.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>288.0</td>\n",
       "      <td>287.0</td>\n",
       "      <td>504.0</td>\n",
       "      <td>...</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>636 rows × 99 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      Pick Number  Team PF  Team Yds  Team Ply  Team Y/P  Team TO  Team FL  \\\n",
       "2020            1    279.0    5169.0    1049.0       4.9     30.0     14.0   \n",
       "2020            2    266.0    4395.0     885.0       5.0     21.0      8.0   \n",
       "2020            3    341.0    5549.0    1021.0       5.4     23.0      8.0   \n",
       "2020            4    341.0    5416.0    1012.0       5.4     33.0     16.0   \n",
       "2020            5    306.0    4960.0    1022.0       4.9     26.0      8.0   \n",
       "...           ...      ...       ...       ...       ...      ...      ...   \n",
       "2001           27    397.0    5961.0     958.0       6.2     28.0     10.0   \n",
       "2001           28    479.0    5776.0    1023.0       5.6     20.0      9.0   \n",
       "2001           29    540.0    7075.0    1014.0       7.0     35.0     12.0   \n",
       "2001           30    429.0    6141.0    1026.0       6.0     29.0     14.0   \n",
       "2001           31    333.0    5014.0    1058.0       4.7     26.0      7.0   \n",
       "\n",
       "      Team 1stD  Team Cmp  Team Att  ...  Def. Rank Yds.2  Def. Rank TD.1  \\\n",
       "2020      312.0     356.0     616.0  ...             32.0            27.0   \n",
       "2020      248.0     298.0     479.0  ...             31.0            17.0   \n",
       "2020      313.0     344.0     571.0  ...             21.0            14.0   \n",
       "2020      311.0     376.0     607.0  ...             20.0            28.0   \n",
       "2020      315.0     371.0     615.0  ...             27.0            21.0   \n",
       "...         ...       ...       ...  ...              ...             ...   \n",
       "2001      319.0     307.0     495.0  ...             15.0            25.0   \n",
       "2001      337.0     284.0     475.0  ...              5.0             5.0   \n",
       "2001      380.0     380.0     587.0  ...             13.0            27.0   \n",
       "2001      357.0     357.0     571.0  ...             25.0            16.0   \n",
       "2001      288.0     287.0     504.0  ...              1.0             1.0   \n",
       "\n",
       "      Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  Def. Rank Start  \\\n",
       "2020           25.0            9.0           27.0             12.0   \n",
       "2020           26.0            4.0           15.0             31.0   \n",
       "2020           10.0            7.0           25.0              3.0   \n",
       "2020            4.0            8.0           29.0             25.0   \n",
       "2020           22.0            1.0           28.0             27.0   \n",
       "...             ...            ...            ...              ...   \n",
       "2001           27.0            6.0           31.0              7.0   \n",
       "2001           14.0           24.0            7.0              1.0   \n",
       "2001           25.0            2.0           22.0             25.0   \n",
       "2001           23.0           12.0           23.0              3.0   \n",
       "2001            1.0           31.0            1.0              2.0   \n",
       "\n",
       "      Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  Def. Rank Pts  \n",
       "2020            18.0             15.0             28.0           22.0  \n",
       "2020            31.0             29.0             29.0           30.0  \n",
       "2020            21.0             28.0             30.0           24.0  \n",
       "2020            19.0             17.0             20.0           25.0  \n",
       "2020            20.0             20.0             26.0           32.0  \n",
       "...              ...              ...              ...            ...  \n",
       "2001            30.0             30.0             30.0           27.0  \n",
       "2001             5.0             15.0             18.0           11.0  \n",
       "2001            15.0             14.0             24.0           30.0  \n",
       "2001            25.0             28.0             25.0           19.0  \n",
       "2001             1.0              1.0              2.0            1.0  \n",
       "\n",
       "[636 rows x 99 columns]"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "num_df = convert_final_df_to_num(specialized_df)\n",
    "num_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "def split_dfs(final_df) -> list:\n",
    "\n",
    "    train_df_years = [*range(CURRENT_YEAR-1, CURRENT_YEAR - YEARS, -1)]\n",
    "\n",
    "    test_df = final_df.loc[[CURRENT_YEAR]]\n",
    "    train_df = final_df.loc[train_df_years]\n",
    "    print(train_df.shape, test_df.shape)\n",
    "\n",
    "    combine = [train_df, test_df]\n",
    "    return combine"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(604, 99) (32, 99)\n"
     ]
    },
    {
     "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>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>Team 1stD</th>\n",
       "      <th>Team Cmp</th>\n",
       "      <th>Team Att</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>225.0</td>\n",
       "      <td>3865.0</td>\n",
       "      <td>902.0</td>\n",
       "      <td>4.3</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>239.0</td>\n",
       "      <td>283.0</td>\n",
       "      <td>495.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>342.0</td>\n",
       "      <td>5769.0</td>\n",
       "      <td>1003.0</td>\n",
       "      <td>5.8</td>\n",
       "      <td>32.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>344.0</td>\n",
       "      <td>331.0</td>\n",
       "      <td>532.0</td>\n",
       "      <td>...</td>\n",
       "      <td>14.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>26.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>333.0</td>\n",
       "      <td>4787.0</td>\n",
       "      <td>971.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>258.0</td>\n",
       "      <td>299.0</td>\n",
       "      <td>524.0</td>\n",
       "      <td>...</td>\n",
       "      <td>26.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>290.0</td>\n",
       "      <td>5379.0</td>\n",
       "      <td>995.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>24.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>300.0</td>\n",
       "      <td>382.0</td>\n",
       "      <td>556.0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5</td>\n",
       "      <td>396.0</td>\n",
       "      <td>6648.0</td>\n",
       "      <td>1055.0</td>\n",
       "      <td>6.3</td>\n",
       "      <td>35.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>388.0</td>\n",
       "      <td>408.0</td>\n",
       "      <td>625.0</td>\n",
       "      <td>...</td>\n",
       "      <td>24.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>29.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 99 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "   Pick Number  Team PF  Team Yds  Team Ply  Team Y/P  Team TO  Team FL  \\\n",
       "0            1    225.0    3865.0     902.0       4.3     28.0     10.0   \n",
       "1            2    342.0    5769.0    1003.0       5.8     32.0     12.0   \n",
       "2            3    333.0    4787.0     971.0       4.9     30.0     11.0   \n",
       "3            4    290.0    5379.0     995.0       5.4     24.0     14.0   \n",
       "4            5    396.0    6648.0    1055.0       6.3     35.0      9.0   \n",
       "\n",
       "   Team 1stD  Team Cmp  Team Att  ...  Def. Rank Yds.2  Def. Rank TD.1  \\\n",
       "0      239.0     283.0     495.0  ...             32.0            32.0   \n",
       "1      344.0     331.0     532.0  ...             14.0            15.0   \n",
       "2      258.0     299.0     524.0  ...             26.0            21.0   \n",
       "3      300.0     382.0     556.0  ...             30.0            21.0   \n",
       "4      388.0     408.0     625.0  ...             24.0            29.0   \n",
       "\n",
       "   Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  Def. Rank Start  \\\n",
       "0           27.0           13.0           26.0             25.0   \n",
       "1            7.0            8.0           32.0             30.0   \n",
       "2           19.0            9.0           21.0             29.0   \n",
       "3           22.0            2.0           23.0             27.0   \n",
       "4           20.0            6.0           28.0             22.0   \n",
       "\n",
       "   Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  Def. Rank Pts  \n",
       "0            28.0             21.0             10.0           22.0  \n",
       "1            16.0             19.0              9.0           26.0  \n",
       "2             1.0              2.0              8.0           19.0  \n",
       "3            18.0             17.0             29.0           32.0  \n",
       "4            10.0              7.0             26.0           29.0  \n",
       "\n",
       "[5 rows x 99 columns]"
      ]
     },
     "execution_count": 35,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "combine = split_dfs(num_df)\n",
    "train_df = combine[0]\n",
    "test_df = combine[1]\n",
    "test_df = test_df.reset_index(drop=True)\n",
    "train_df = train_df.reset_index(drop=True)\n",
    "train_df.head()\n",
    "\n",
    "# test_df.to_csv('Test_DF.csv', encoding='utf-8-sig', index=True)\n",
    "# train_df.to_csv('Train_DF.csv', encoding = 'utf-8-sig', index=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "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>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>Team Y/P</th>\n",
       "      <th>Team TO</th>\n",
       "      <th>Team FL</th>\n",
       "      <th>Team 1stD</th>\n",
       "      <th>Team Cmp</th>\n",
       "      <th>Team Att</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>1</td>\n",
       "      <td>225.0</td>\n",
       "      <td>3865.0</td>\n",
       "      <td>902.0</td>\n",
       "      <td>4.3</td>\n",
       "      <td>28.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>239.0</td>\n",
       "      <td>283.0</td>\n",
       "      <td>495.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>2</td>\n",
       "      <td>342.0</td>\n",
       "      <td>5769.0</td>\n",
       "      <td>1003.0</td>\n",
       "      <td>5.8</td>\n",
       "      <td>32.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>344.0</td>\n",
       "      <td>331.0</td>\n",
       "      <td>532.0</td>\n",
       "      <td>...</td>\n",
       "      <td>14.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>26.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>3</td>\n",
       "      <td>333.0</td>\n",
       "      <td>4787.0</td>\n",
       "      <td>971.0</td>\n",
       "      <td>4.9</td>\n",
       "      <td>30.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>258.0</td>\n",
       "      <td>299.0</td>\n",
       "      <td>524.0</td>\n",
       "      <td>...</td>\n",
       "      <td>26.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>4</td>\n",
       "      <td>290.0</td>\n",
       "      <td>5379.0</td>\n",
       "      <td>995.0</td>\n",
       "      <td>5.4</td>\n",
       "      <td>24.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>300.0</td>\n",
       "      <td>382.0</td>\n",
       "      <td>556.0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>5</td>\n",
       "      <td>396.0</td>\n",
       "      <td>6648.0</td>\n",
       "      <td>1055.0</td>\n",
       "      <td>6.3</td>\n",
       "      <td>35.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>388.0</td>\n",
       "      <td>408.0</td>\n",
       "      <td>625.0</td>\n",
       "      <td>...</td>\n",
       "      <td>24.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>29.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 99 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    Pick Number  Team PF  Team Yds  Team Ply  Team Y/P  Team TO  Team FL  \\\n",
       "32            1    225.0    3865.0     902.0       4.3     28.0     10.0   \n",
       "33            2    342.0    5769.0    1003.0       5.8     32.0     12.0   \n",
       "34            3    333.0    4787.0     971.0       4.9     30.0     11.0   \n",
       "35            4    290.0    5379.0     995.0       5.4     24.0     14.0   \n",
       "36            5    396.0    6648.0    1055.0       6.3     35.0      9.0   \n",
       "\n",
       "    Team 1stD  Team Cmp  Team Att  ...  Def. Rank Yds.2  Def. Rank TD.1  \\\n",
       "32      239.0     283.0     495.0  ...             32.0            32.0   \n",
       "33      344.0     331.0     532.0  ...             14.0            15.0   \n",
       "34      258.0     299.0     524.0  ...             26.0            21.0   \n",
       "35      300.0     382.0     556.0  ...             30.0            21.0   \n",
       "36      388.0     408.0     625.0  ...             24.0            29.0   \n",
       "\n",
       "    Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  Def. Rank Start  \\\n",
       "32           27.0           13.0           26.0             25.0   \n",
       "33            7.0            8.0           32.0             30.0   \n",
       "34           19.0            9.0           21.0             29.0   \n",
       "35           22.0            2.0           23.0             27.0   \n",
       "36           20.0            6.0           28.0             22.0   \n",
       "\n",
       "    Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  Def. Rank Pts  \n",
       "32            28.0             21.0             10.0           22.0  \n",
       "33            16.0             19.0              9.0           26.0  \n",
       "34             1.0              2.0              8.0           19.0  \n",
       "35            18.0             17.0             29.0           32.0  \n",
       "36            10.0              7.0             26.0           29.0  \n",
       "\n",
       "[5 rows x 99 columns]"
      ]
     },
     "execution_count": 36,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_df.index = train_df.index + 32\n",
    "train_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [],
   "source": [
    "# DATA PREPARATION FUNCTIONS\n",
    "\n",
    "def convert_df_to_weighted_df(df):\n",
    "    std_list = list(df.std(axis = 0, skipna= True))\n",
    "    mean_list = list(df.mean(axis=0, skipna= True))\n",
    "    frames = []\n",
    "    for i in range(df.index[0], df.index[0]+len(df.index)):\n",
    "        row_list = list(df.loc[i])\n",
    "        weighted_row = []\n",
    "        for j in range(len(row_list)):\n",
    "            weighted_row.append((row_list[j] - mean_list[j])/std_list[j])\n",
    "        weighted_df_row = pd.DataFrame([weighted_row], columns=df.columns)\n",
    "        frames.append(weighted_df_row)\n",
    "    weighted_df = pd.concat(frames)\n",
    "    weighted_df.index = df.index\n",
    "    return weighted_df\n",
    "\n",
    "def ready_dataset_for_modeling(train_df, test_df) -> list:\n",
    "    weighted_train_df = convert_df_to_weighted_df(train_df)\n",
    "    weighted_test_df = convert_df_to_weighted_df(test_df)\n",
    "    \n",
    "    train_dataset = weighted_train_df.to_numpy().tolist()\n",
    "    test_dataset = weighted_test_df.to_numpy().tolist()\n",
    "    \n",
    "    combine = [train_dataset, test_dataset]\n",
    "    return combine"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'def get_model_ready_combine(final_df):\\n    clean_final_df = clean_final_df(final_df)\\n    specialized_df = specialize_final_df(clean_final_df)\\n    reference_df = specialized_df.copy(deep=True)\\n    num_df = convert_final_df_to_num(specialized_df)\\n    combine = split_dfs(num_df)\\n    train_df = combine[0]\\n    test_df = combine[1]\\n    test_df = test_df.reset_index(drop=True)\\n    train_df = train_df.reset_index(drop=True)\\n    train_df.index = train_df.index + 32\\n    return ready_dataset_for_modeling(train_df, test_df)\\n    \\nmodel_ready_combine = get_model_ready_combine(final_df)\\ntrain_dataset = model_ready_combine[0]\\ntest_dataset = model_ready_combine[1]\\ntrain_dataset\\n    '"
      ]
     },
     "execution_count": 38,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "'''def get_model_ready_combine(final_df):\n",
    "    clean_final_df = clean_final_df(final_df)\n",
    "    specialized_df = specialize_final_df(clean_final_df)\n",
    "    reference_df = specialized_df.copy(deep=True)\n",
    "    num_df = convert_final_df_to_num(specialized_df)\n",
    "    combine = split_dfs(num_df)\n",
    "    train_df = combine[0]\n",
    "    test_df = combine[1]\n",
    "    test_df = test_df.reset_index(drop=True)\n",
    "    train_df = train_df.reset_index(drop=True)\n",
    "    train_df.index = train_df.index + 32\n",
    "    return ready_dataset_for_modeling(train_df, test_df)\n",
    "    \n",
    "model_ready_combine = get_model_ready_combine(final_df)\n",
    "train_dataset = model_ready_combine[0]\n",
    "test_dataset = model_ready_combine[1]\n",
    "train_dataset\n",
    "    '''"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {},
   "outputs": [],
   "source": [
    "model_ready_combine = ready_dataset_for_modeling(train_df, test_df)\n",
    "train_dataset = model_ready_combine[0]\n",
    "test_dataset = model_ready_combine[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "metadata": {},
   "outputs": [],
   "source": [
    "# K NEAREST NEIGHBORS MODEL FUNCTIONS\n",
    "\n",
    "def euclidean_distance(row1, row2):\n",
    "    distance = 0.0\n",
    "    for i in range(len(row1)-1):\n",
    "        distance += (row1[i] - row2[i])**2\n",
    "    return math.sqrt(distance)\n",
    "\n",
    "def get_neighbors(train, test_row, num_neighbors):\n",
    "    distances = list()\n",
    "    for i in range(0, len(train)-1):\n",
    "        index = i+32\n",
    "        train_row = train[i]\n",
    "        dist = euclidean_distance(test_row, train_row)\n",
    "        distances.append((train_row, dist, index))\n",
    "    distances.sort(key=lambda tup: tup[1])\n",
    "    neighbors = list()\n",
    "    for i in range(num_neighbors):\n",
    "        neighbors.append((distances[i][2], round(distances[i][1], 2)))\n",
    "    return neighbors\n",
    "\n",
    "def prediction_data(train_dataset, test_dataset, test_row_index) -> 'DataFrame':\n",
    "    neighbors = get_neighbors(train_dataset, test_dataset[test_row_index], NUM_NEIGHBORS)\n",
    "    \n",
    "    frames = []\n",
    "    index_col = [test_row_index]\n",
    "    distance_col = [0]\n",
    "    \n",
    "    for neighbor in neighbors:\n",
    "        index_col.append(neighbor[0])\n",
    "        distance_col.append(neighbor[1])\n",
    "        frames.append(pd.DataFrame([list(reference_df.loc[int(neighbor[0])])], columns=reference_df.columns))\n",
    "        \n",
    "    big_df = pd.concat(frames)\n",
    "    big_df = pd.DataFrame([reference_df.loc[test_row_index]], columns=reference_df.columns).append(big_df, ignore_index=True)\n",
    "    big_df.insert(0, 'Distance', distance_col, True)\n",
    "    big_df.insert(0, 'Index', index_col, True)\n",
    "    big_df.insert(2, 'Unit', big_df['Position Drafted'].map(UNIT_DICT), True)\n",
    "    \n",
    "    return big_df\n",
    "\n",
    "def unit_prediction(train_dataset, test_dataset, test_row_index):\n",
    "    predict_dict = {}\n",
    "    neighbors = get_neighbors(train_dataset, test_dataset[test_row_index], NUM_NEIGHBORS)\n",
    "    \n",
    "    sum = 0\n",
    "    for neighbor in neighbors:\n",
    "        position = list(reference_df.loc[int(neighbor[0])])[0]\n",
    "        for i in UNIT_DICT.keys():\n",
    "            if position == i:\n",
    "                unit = UNIT_DICT[position]\n",
    "        if unit not in predict_dict:\n",
    "            predict_dict[unit] = 0\n",
    "        predict_dict[unit] += 1\n",
    "        sum += 1\n",
    "        \n",
    "    #print('Actual Unit: ' + UNIT_DICT[list(reference_df.loc[test_row_index])[0]])\n",
    "    #for i in predict_dict:\n",
    "        #print(i, 'Count:'+str(predict_dict[i]), str(round(predict_dict[i]/sum, 4)*100) + '%')\n",
    "    \n",
    "    percent_dict = {'OFF':0,'DEF':0}\n",
    "    for i in predict_dict:\n",
    "        percent_dict[i] = round(predict_dict[i]/sum, 4)*100\n",
    "    percent_dict['Actual Unit'] = UNIT_DICT[list(reference_df.loc[test_row_index])[0]]\n",
    "    return percent_dict"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 41,
   "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>Index</th>\n",
       "      <th>Distance</th>\n",
       "      <th>Unit</th>\n",
       "      <th>Position Drafted</th>\n",
       "      <th>Team</th>\n",
       "      <th>Draft Year</th>\n",
       "      <th>Pick Number</th>\n",
       "      <th>Team PF</th>\n",
       "      <th>Team Yds</th>\n",
       "      <th>Team Ply</th>\n",
       "      <th>...</th>\n",
       "      <th>Def. Rank Yds.2</th>\n",
       "      <th>Def. Rank TD.1</th>\n",
       "      <th>Def. Rank Y/A</th>\n",
       "      <th>Def. Rank Sc%</th>\n",
       "      <th>Def. Rank TO%</th>\n",
       "      <th>Def. Rank Start</th>\n",
       "      <th>Def. Rank Time</th>\n",
       "      <th>Def. Rank Plays</th>\n",
       "      <th>Def. Rank Yds.4</th>\n",
       "      <th>Def. Rank Pts</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>0.00</td>\n",
       "      <td>OFF</td>\n",
       "      <td>QB</td>\n",
       "      <td>Bengals</td>\n",
       "      <td>2020</td>\n",
       "      <td>1</td>\n",
       "      <td>279.0</td>\n",
       "      <td>5169.0</td>\n",
       "      <td>1049.0</td>\n",
       "      <td>...</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>605</td>\n",
       "      <td>7.02</td>\n",
       "      <td>OFF</td>\n",
       "      <td>QB</td>\n",
       "      <td>Falcons</td>\n",
       "      <td>2001</td>\n",
       "      <td>1</td>\n",
       "      <td>252.0</td>\n",
       "      <td>3994.0</td>\n",
       "      <td>925.0</td>\n",
       "      <td>...</td>\n",
       "      <td>27.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>25.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>558</td>\n",
       "      <td>7.07</td>\n",
       "      <td>OFF</td>\n",
       "      <td>WR</td>\n",
       "      <td>Cardinals</td>\n",
       "      <td>2003</td>\n",
       "      <td>17</td>\n",
       "      <td>262.0</td>\n",
       "      <td>4563.0</td>\n",
       "      <td>1003.0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>559</td>\n",
       "      <td>7.10</td>\n",
       "      <td>DEF</td>\n",
       "      <td>DE</td>\n",
       "      <td>Cardinals</td>\n",
       "      <td>2003</td>\n",
       "      <td>18</td>\n",
       "      <td>262.0</td>\n",
       "      <td>4563.0</td>\n",
       "      <td>1003.0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>256</td>\n",
       "      <td>7.48</td>\n",
       "      <td>OFF</td>\n",
       "      <td>QB</td>\n",
       "      <td>Redskins</td>\n",
       "      <td>2012</td>\n",
       "      <td>2</td>\n",
       "      <td>288.0</td>\n",
       "      <td>5387.0</td>\n",
       "      <td>1032.0</td>\n",
       "      <td>...</td>\n",
       "      <td>18.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>512</td>\n",
       "      <td>7.67</td>\n",
       "      <td>OFF</td>\n",
       "      <td>WR</td>\n",
       "      <td>Cardinals</td>\n",
       "      <td>2004</td>\n",
       "      <td>3</td>\n",
       "      <td>225.0</td>\n",
       "      <td>4490.0</td>\n",
       "      <td>981.0</td>\n",
       "      <td>...</td>\n",
       "      <td>19.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>31.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>6 rows × 105 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "   Index  Distance Unit Position Drafted       Team  Draft Year  Pick Number  \\\n",
       "0      0      0.00  OFF               QB    Bengals        2020            1   \n",
       "1    605      7.02  OFF               QB    Falcons        2001            1   \n",
       "2    558      7.07  OFF               WR  Cardinals        2003           17   \n",
       "3    559      7.10  DEF               DE  Cardinals        2003           18   \n",
       "4    256      7.48  OFF               QB   Redskins        2012            2   \n",
       "5    512      7.67  OFF               WR  Cardinals        2004            3   \n",
       "\n",
       "   Team PF  Team Yds  Team Ply  ...  Def. Rank Yds.2  Def. Rank TD.1  \\\n",
       "0    279.0    5169.0    1049.0  ...             32.0            27.0   \n",
       "1    252.0    3994.0     925.0  ...             27.0            24.0   \n",
       "2    262.0    4563.0    1003.0  ...             30.0            23.0   \n",
       "3    262.0    4563.0    1003.0  ...             30.0            23.0   \n",
       "4    288.0    5387.0    1032.0  ...             18.0            23.0   \n",
       "5    225.0    4490.0     981.0  ...             19.0            24.0   \n",
       "\n",
       "   Def. Rank Y/A  Def. Rank Sc%  Def. Rank TO%  Def. Rank Start  \\\n",
       "0           25.0            9.0           27.0             12.0   \n",
       "1           24.0            7.0           20.0             10.0   \n",
       "2           23.0            6.0           26.0             18.0   \n",
       "3           23.0            6.0           26.0             18.0   \n",
       "4           18.0            5.0           25.0             27.0   \n",
       "5           15.0            3.0           21.0             29.0   \n",
       "\n",
       "   Def. Rank Time  Def. Rank Plays  Def. Rank Yds.4  Def. Rank Pts  \n",
       "0            18.0             15.0             28.0           22.0  \n",
       "1            20.0             21.0             26.0           25.0  \n",
       "2            25.0             26.0             29.0           29.0  \n",
       "3            25.0             26.0             29.0           29.0  \n",
       "4            16.0             10.0             15.0           24.0  \n",
       "5            24.0             21.0             26.0           31.0  \n",
       "\n",
       "[6 rows x 105 columns]"
      ]
     },
     "execution_count": 41,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "PICK_NUM = 1\n",
    "test = prediction_data(train_dataset, test_dataset, PICK_NUM-1)\n",
    "test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'OFF': 80.0, 'DEF': 20.0, 'Actual Unit': 'OFF'}"
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "unit_prediction(train_dataset, test_dataset, PICK_NUM-1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 56,
   "metadata": {},
   "outputs": [],
   "source": [
    "# TESTING FUNCTIONS\n",
    "\n",
    "def test_unit_prediction(train_dataset, test_dataset):\n",
    "    correct_predictions = 0\n",
    "    length = len(test_dataset)\n",
    "    for i in range(length):\n",
    "        predict_dict = unit_prediction(train_dataset, test_dataset, i)\n",
    "        offense = predict_dict['OFF']\n",
    "        defense = predict_dict['DEF']\n",
    "        if predict_dict['Actual Unit'] == 'OFF' and offense > defense:\n",
    "            correct_predictions += 1\n",
    "        elif predict_dict['Actual Unit'] == 'DEF' and defense > offense:\n",
    "            correct_predictions += 1\n",
    "    return round(correct_predictions/length, 4)\n",
    "\n",
    "def get_influence_of_all_columns(train_df, test_df):\n",
    "    remove_col_dict = {}\n",
    "    for i in train_df.columns:\n",
    "        new_train_df = train_df.drop(columns=i)\n",
    "        test_df = test_df.drop(columns=i)\n",
    "        model_ready_combine = ready_dataset_for_modeling(new_train_df, test_df)\n",
    "        train_dataset = model_ready_combine[0]\n",
    "        test_dataset = model_ready_combine[1]\n",
    "        prediction = test_unit_prediction(train_dataset, test_dataset)\n",
    "        remove_col_dict[i] = prediction\n",
    "        print(i, remove_col_dict[i])\n",
    "        \n",
    "    return remove_col_dict\n",
    "\n",
    "def isolate_influential_columns(columns_influence):\n",
    "    good_cols = []\n",
    "    for i in columns_influence:\n",
    "        if columns_influence[i] < .4:\n",
    "            good_cols.append(i)\n",
    "    return good_cols\n",
    "\n",
    "def isolate_noninfluential_columns(train_df, test_df, columns_influence):\n",
    "    drop_cols = []\n",
    "    for i in columns_influence:\n",
    "        if columns_influence[i] > .5:\n",
    "            drop_cols.append(i)\n",
    "    new_train_df = train_df.drop(columns=drop_cols)\n",
    "    test_df = test_df.drop(columns=drop_cols)\n",
    "    model_ready_combine = ready_dataset_for_modeling(new_train_df, test_df)\n",
    "    train_dataset = model_ready_combine[0]\n",
    "    test_dataset = model_ready_combine[1]\n",
    "    print(drop_cols)\n",
    "    combine = [train_dataset, test_dataset]\n",
    "    return combine\n",
    "\n",
    "def test_num_correct_predictions(train_dataset, test_dataset):\n",
    "    correct_predictions = 0\n",
    "    length = len(test_dataset)\n",
    "    for i in range(length):\n",
    "        predict_dict = unit_prediction(train_dataset, test_dataset, i)\n",
    "        offense = predict_dict['OFF']\n",
    "        defense = predict_dict['DEF']\n",
    "        if predict_dict['Actual Unit'] == 'OFF':\n",
    "            correct_predictions += offense\n",
    "        elif predict_dict['Actual Unit'] == 'DEF':\n",
    "            correct_predictions += defense\n",
    "    return round(correct_predictions/length, 4)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.5312"
      ]
     },
     "execution_count": 49,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_unit_prediction(train_dataset, test_dataset)\n",
    "# All Columns Accuracy: .5312\n",
    "# No Rank Columns Accuracy: .5\n",
    "# No Defensive Rank Col Accuracy: .5625\n",
    "# No Offensive Rank Col Accuracy: .5625\n",
    "# No Columns Influence > .4:  .5312\n",
    "# No Columns Influence > .5: .5"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "53.125"
      ]
     },
     "execution_count": 50,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_num_correct_predictions(train_dataset, test_dataset)\n",
    "# All Columns Accuracy: 53.75\n",
    "# No Columns Influence > .4: 51.875\n",
    "# No Columns Influence > .5: 52.1875"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Pick Number 0.5312\n",
      "Team PF 0.4688\n",
      "Team Yds 0.5312\n",
      "Team Ply 0.6562\n",
      "Team Y/P 0.5312\n",
      "Team TO 0.5312\n",
      "Team FL 0.5312\n",
      "Team 1stD 0.5312\n",
      "Team Cmp 0.5\n",
      "Team Att 0.3438\n",
      "Team Yds.1 0.5\n",
      "Team TD 0.4688\n",
      "Team Int 0.5312\n",
      "Team NY/A 0.5312\n",
      "Team 1stD.1 0.4688\n",
      "Team Att.1 0.4688\n",
      "Team Yds.2 0.4375\n",
      "Team TD.1 0.4688\n",
      "Team Y/A 0.5625\n",
      "Team 1stD.2 0.4375\n",
      "Team Pen 0.5625\n",
      "Team Yds.3 0.5312\n",
      "Team 1stPy 0.5\n",
      "Team #Dr 0.5625\n",
      "Team Sc% 0.5938\n",
      "Team TO% 0.375\n",
      "Team Start 0.4375\n",
      "Team Time 0.5625\n",
      "Team Plays 0.6562\n",
      "Team Yds.4 0.4062\n",
      "Team Pts 0.5938\n",
      "Opp. PF 0.5625\n",
      "Opp. Yds 0.375\n",
      "Opp. Ply 0.6875\n",
      "Opp. Y/P 0.4062\n",
      "Opp. TO 0.5\n",
      "Opp. FL 0.5625\n",
      "Opp. 1stD 0.5625\n",
      "Opp. Cmp 0.625\n",
      "Opp. Att 0.6875\n",
      "Opp. Yds.1 0.5312\n",
      "Opp. TD 0.5312\n",
      "Opp. Int 0.5938\n",
      "Opp. NY/A 0.5\n",
      "Opp. 1stD.1 0.5625\n",
      "Opp. Att.1 0.4375\n",
      "Opp. Yds.2 0.5\n",
      "Opp. TD.1 0.75\n",
      "Opp. Y/A 0.4062\n",
      "Opp. 1stD.2 0.5625\n",
      "Opp. Pen 0.4688\n",
      "Opp. Yds.3 0.4688\n",
      "Opp. 1stPy 0.5312\n",
      "Opp. #Dr 0.5\n",
      "Opp. Sc% 0.4375\n",
      "Opp. TO% 0.4062\n",
      "Opp. Start 0.5625\n",
      "Opp. Time 0.5312\n",
      "Opp. Plays 0.4375\n",
      "Opp. Yds.4 0.4062\n",
      "Opp. Pts 0.4062\n",
      "Off. Rank PF 0.4688\n",
      "Off. Rank TO 0.4688\n",
      "Off. Rank FL 0.4688\n",
      "Off. Rank Att 0.5938\n",
      "Off. Rank Yds.1 0.5938\n",
      "Off. Rank TD 0.4688\n",
      "Off. Rank Int 0.5625\n",
      "Off. Rank NY/A 0.5\n",
      "Off. Rank Att.1 0.4375\n",
      "Off. Rank Yds.2 0.4062\n",
      "Off. Rank TD.1 0.5\n",
      "Off. Rank Y/A 0.5\n",
      "Off. Rank Sc% 0.5625\n",
      "Off. Rank TO% 0.5\n",
      "Off. Rank Start 0.4688\n",
      "Off. Rank Time 0.4688\n",
      "Off. Rank Plays 0.4688\n",
      "Off. Rank Yds.4 0.375\n",
      "Off. Rank Pts 0.5625\n",
      "Def. Rank PF 0.5\n",
      "Def. Rank TO 0.4062\n",
      "Def. Rank FL 0.5\n",
      "Def. Rank Att 0.5625\n",
      "Def. Rank Yds.1 0.5625\n",
      "Def. Rank TD 0.5\n",
      "Def. Rank Int 0.4375\n",
      "Def. Rank NY/A 0.4688\n",
      "Def. Rank Att.1 0.5\n",
      "Def. Rank Yds.2 0.5938\n",
      "Def. Rank TD.1 0.5938\n",
      "Def. Rank Y/A 0.5312\n",
      "Def. Rank Sc% 0.5\n",
      "Def. Rank TO% 0.4688\n",
      "Def. Rank Start 0.4062\n",
      "Def. Rank Time 0.625\n",
      "Def. Rank Plays 0.4062\n",
      "Def. Rank Yds.4 0.4375\n",
      "Def. Rank Pts 0.4375\n"
     ]
    }
   ],
   "source": [
    "# Takes 5 minutes to run\n",
    "columns_influence = get_influence_of_all_columns(train_df, test_df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 57,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Team Att', 'Team TO%', 'Opp. Yds', 'Off. Rank Yds.4']"
      ]
     },
     "execution_count": 57,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "isolate_influential_columns(columns_influence)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "53.75"
      ]
     },
     "execution_count": 26,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# NORM DATASET\n",
    "model_ready_combine = ready_dataset_for_modeling(train_df, test_df)\n",
    "train_dataset = model_ready_combine[0]\n",
    "test_dataset = model_ready_combine[1]\n",
    "test_num_correct_predictions(train_dataset, test_dataset)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 55,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['Pick Number', 'Team 1stD', 'Team TD.1', 'Team Time', 'Opp. FL', 'Opp. Att', 'Opp. TD', 'Opp. Int', 'Opp. TD.1', 'Off. Rank Pts', 'Def. Rank FL']\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "52.5"
      ]
     },
     "execution_count": 55,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# DROP ISOLATED COLS DATASET\n",
    "combine = isolate_influential_columns(train_df, test_df, columns_influence)\n",
    "train_dataset = combine[0]\n",
    "test_dataset = combine[1]\n",
    "test_num_correct_predictions(train_dataset, test_dataset)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 75,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.5312"
      ]
     },
     "execution_count": 75,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# DROP ANY COLS DATASET\n",
    "drop_cols = ['Off. Rank Time']\n",
    "new_train_df = train_df.drop(columns=drop_cols)\n",
    "new_test_df = test_df.drop(columns=drop_cols)\n",
    "model_ready_combine = ready_dataset_for_modeling(new_train_df, new_test_df)\n",
    "train_dataset = model_ready_combine[0]\n",
    "test_dataset = model_ready_combine[1]\n",
    "test_unit_prediction(train_dataset, test_dataset)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}