Decision_Tree_Regresssssion.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pandas as pd\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>Additional_Info</th>\n",
       "      <th>Price</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>BLR → DEL</td>\n",
       "      <td>22:20</td>\n",
       "      <td>01:10 22 Mar</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>No info</td>\n",
       "      <td>3897</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Air India</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → IXR → BBI → BLR</td>\n",
       "      <td>05:50</td>\n",
       "      <td>13:15</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>No info</td>\n",
       "      <td>7662</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → LKO → BOM → COK</td>\n",
       "      <td>09:25</td>\n",
       "      <td>04:25 10 Jun</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>No info</td>\n",
       "      <td>13882</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → NAG → BLR</td>\n",
       "      <td>18:05</td>\n",
       "      <td>23:30</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>No info</td>\n",
       "      <td>6218</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>BLR → NAG → DEL</td>\n",
       "      <td>16:50</td>\n",
       "      <td>21:35</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>No info</td>\n",
       "      <td>13302</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Airline Date_of_Journey    Source Destination                  Route  \\\n",
       "0       IndiGo      24/03/2019  Banglore   New Delhi              BLR → DEL   \n",
       "1    Air India       1/05/2019   Kolkata    Banglore  CCU → IXR → BBI → BLR   \n",
       "2  Jet Airways       9/06/2019     Delhi      Cochin  DEL → LKO → BOM → COK   \n",
       "3       IndiGo      12/05/2019   Kolkata    Banglore        CCU → NAG → BLR   \n",
       "4       IndiGo      01/03/2019  Banglore   New Delhi        BLR → NAG → DEL   \n",
       "\n",
       "  Dep_Time  Arrival_Time Duration Total_Stops Additional_Info  Price  \n",
       "0    22:20  01:10 22 Mar   2h 50m    non-stop         No info   3897  \n",
       "1    05:50         13:15   7h 25m     2 stops         No info   7662  \n",
       "2    09:25  04:25 10 Jun      19h     2 stops         No info  13882  \n",
       "3    18:05         23:30   5h 25m      1 stop         No info   6218  \n",
       "4    16:50         21:35   4h 45m      1 stop         No info  13302  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = pd.read_excel(\"Flight_Ticket_Participant_Datasets/Data_Train.xlsx\")\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>Price</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>BLR → DEL</td>\n",
       "      <td>22:20</td>\n",
       "      <td>01:10 22 Mar</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>3897</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Air India</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → IXR → BBI → BLR</td>\n",
       "      <td>05:50</td>\n",
       "      <td>13:15</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>7662</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → LKO → BOM → COK</td>\n",
       "      <td>09:25</td>\n",
       "      <td>04:25 10 Jun</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>13882</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → NAG → BLR</td>\n",
       "      <td>18:05</td>\n",
       "      <td>23:30</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>6218</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>BLR → NAG → DEL</td>\n",
       "      <td>16:50</td>\n",
       "      <td>21:35</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>13302</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Airline Date_of_Journey    Source Destination                  Route  \\\n",
       "0       IndiGo      24/03/2019  Banglore   New Delhi              BLR → DEL   \n",
       "1    Air India       1/05/2019   Kolkata    Banglore  CCU → IXR → BBI → BLR   \n",
       "2  Jet Airways       9/06/2019     Delhi      Cochin  DEL → LKO → BOM → COK   \n",
       "3       IndiGo      12/05/2019   Kolkata    Banglore        CCU → NAG → BLR   \n",
       "4       IndiGo      01/03/2019  Banglore   New Delhi        BLR → NAG → DEL   \n",
       "\n",
       "  Dep_Time  Arrival_Time Duration Total_Stops  Price  \n",
       "0    22:20  01:10 22 Mar   2h 50m    non-stop   3897  \n",
       "1    05:50         13:15   7h 25m     2 stops   7662  \n",
       "2    09:25  04:25 10 Jun      19h     2 stops  13882  \n",
       "3    18:05         23:30   5h 25m      1 stop   6218  \n",
       "4    16:50         21:35   4h 45m      1 stop  13302  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = data.drop([\"Additional_Info\"],axis=1)\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>22:20</td>\n",
       "      <td>01:10 22 Mar</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Air India</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>05:50</td>\n",
       "      <td>13:15</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>09:25</td>\n",
       "      <td>04:25 10 Jun</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>18:05</td>\n",
       "      <td>23:30</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>16:50</td>\n",
       "      <td>21:35</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Airline Date_of_Journey    Source Destination Dep_Time  Arrival_Time  \\\n",
       "0       IndiGo      24/03/2019  Banglore   New Delhi    22:20  01:10 22 Mar   \n",
       "1    Air India       1/05/2019   Kolkata    Banglore    05:50         13:15   \n",
       "2  Jet Airways       9/06/2019     Delhi      Cochin    09:25  04:25 10 Jun   \n",
       "3       IndiGo      12/05/2019   Kolkata    Banglore    18:05         23:30   \n",
       "4       IndiGo      01/03/2019  Banglore   New Delhi    16:50         21:35   \n",
       "\n",
       "  Duration Total_Stops  \n",
       "0   2h 50m    non-stop  \n",
       "1   7h 25m     2 stops  \n",
       "2      19h     2 stops  \n",
       "3   5h 25m      1 stop  \n",
       "4   4h 45m      1 stop  "
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "label = data[\"Price\"]\n",
    "# label.head()\n",
    "data = data.drop([\"Route\",\"Price\"],axis=1)\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0    24/03/2019\n",
       "1     1/05/2019\n",
       "2     9/06/2019\n",
       "3    12/05/2019\n",
       "4    01/03/2019\n",
       "Name: Date_of_Journey, dtype: object"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "temp_date = data[\"Date_of_Journey\"]\n",
    "temp_date.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "from datetime import date\n",
    "no_of_days = []\n",
    "weekday = []\n",
    "\n",
    "for i in temp_date:\n",
    "    s = i.split(\"/\")\n",
    "    year = int(s[2])\n",
    "    month = int(s[1])\n",
    "    day = int(s[0])\n",
    "    \n",
    "    x = date(year,month,day)\n",
    "    y = date.today()\n",
    "    \n",
    "    weekday.append(x.weekday())\n",
    "    \n",
    "    no_of_days.append((x-y).days)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>22:20</td>\n",
       "      <td>01:10 22 Mar</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>17</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Air India</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>05:50</td>\n",
       "      <td>13:15</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>55</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>09:25</td>\n",
       "      <td>04:25 10 Jun</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>94</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>18:05</td>\n",
       "      <td>23:30</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>New Delhi</td>\n",
       "      <td>16:50</td>\n",
       "      <td>21:35</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>-6</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Airline Date_of_Journey    Source Destination Dep_Time  Arrival_Time  \\\n",
       "0       IndiGo      24/03/2019  Banglore   New Delhi    22:20  01:10 22 Mar   \n",
       "1    Air India       1/05/2019   Kolkata    Banglore    05:50         13:15   \n",
       "2  Jet Airways       9/06/2019     Delhi      Cochin    09:25  04:25 10 Jun   \n",
       "3       IndiGo      12/05/2019   Kolkata    Banglore    18:05         23:30   \n",
       "4       IndiGo      01/03/2019  Banglore   New Delhi    16:50         21:35   \n",
       "\n",
       "  Duration Total_Stops  No_of_days  Weekday  \n",
       "0   2h 50m    non-stop          17        6  \n",
       "1   7h 25m     2 stops          55        2  \n",
       "2      19h     2 stops          94        6  \n",
       "3   5h 25m      1 stop          66        6  \n",
       "4   4h 45m      1 stop          -6        4  "
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data[\"No_of_days\"] = no_of_days\n",
    "data[\"Weekday\"] = weekday\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>3</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>22:20</td>\n",
       "      <td>01:10 22 Mar</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>17</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>05:50</td>\n",
       "      <td>13:15</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>55</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>09:25</td>\n",
       "      <td>04:25 10 Jun</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>94</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>18:05</td>\n",
       "      <td>23:30</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>16:50</td>\n",
       "      <td>21:35</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>-6</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline Date_of_Journey  Source  Destination Dep_Time  Arrival_Time  \\\n",
       "0        3      24/03/2019       0            5    22:20  01:10 22 Mar   \n",
       "1        1       1/05/2019       3            0    05:50         13:15   \n",
       "2        4       9/06/2019       2            1    09:25  04:25 10 Jun   \n",
       "3        3      12/05/2019       3            0    18:05         23:30   \n",
       "4        3      01/03/2019       0            5    16:50         21:35   \n",
       "\n",
       "  Duration Total_Stops  No_of_days  Weekday  \n",
       "0   2h 50m    non-stop          17        6  \n",
       "1   7h 25m     2 stops          55        2  \n",
       "2      19h     2 stops          94        6  \n",
       "3   5h 25m      1 stop          66        6  \n",
       "4   4h 45m      1 stop          -6        4  "
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from sklearn import preprocessing\n",
    "\n",
    "le = preprocessing.LabelEncoder()\n",
    "\n",
    "\n",
    "le.fit(data[\"Airline\"])\n",
    "data[\"Airline\"] = le.transform(data[\"Airline\"])\n",
    "\n",
    "le.fit(data[\"Source\"])\n",
    "data[\"Source\"] = le.transform(data[\"Source\"])\n",
    "\n",
    "le.fit(data[\"Destination\"])\n",
    "data[\"Destination\"] = le.transform(data[\"Destination\"])\n",
    "\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "day = data[\"Weekday\"]\n",
    "expensive_ticket_day = []\n",
    "\n",
    "for d in day:\n",
    "    if d==4 or d==5:\n",
    "        expensive_ticket_day.append(1)\n",
    "    else:\n",
    "        expensive_ticket_day.append(0)\n",
    "        \n",
    "data[\"Expensive_Day\"] = expensive_ticket_day"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "      <th>Expensive_Day</th>\n",
       "      <th>Arrival_Time</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>3</td>\n",
       "      <td>24/03/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>22:20</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>17</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>01:10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>1/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>05:50</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>55</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>13:15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>9/06/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>09:25</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>94</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>04:25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>18:05</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>23:30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>01/03/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>16:50</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>-6</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>21:35</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline Date_of_Journey  Source  Destination Dep_Time Duration Total_Stops  \\\n",
       "0        3      24/03/2019       0            5    22:20   2h 50m    non-stop   \n",
       "1        1       1/05/2019       3            0    05:50   7h 25m     2 stops   \n",
       "2        4       9/06/2019       2            1    09:25      19h     2 stops   \n",
       "3        3      12/05/2019       3            0    18:05   5h 25m      1 stop   \n",
       "4        3      01/03/2019       0            5    16:50   4h 45m      1 stop   \n",
       "\n",
       "   No_of_days  Weekday  Expensive_Day Arrival_Time  \n",
       "0          17        6              0        01:10  \n",
       "1          55        2              0        13:15  \n",
       "2          94        6              0        04:25  \n",
       "3          66        6              0        23:30  \n",
       "4          -6        4              1        21:35  "
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "arrival_time = data[\"Arrival_Time\"]\n",
    "temp = []\n",
    "\n",
    "for i in arrival_time:\n",
    "    temp.append(i[:5])\n",
    "    \n",
    "data = data.drop([\"Arrival_Time\"],axis=1)\n",
    "data[\"Arrival_Time\"] = temp\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "overnight_flight = []\n",
    "dep = data[\"Dep_Time\"]\n",
    "arr = data[\"Arrival_Time\"]\n",
    "for i,j in zip(dep,arr):\n",
    "    if i>j:\n",
    "        overnight_flight.append(1)\n",
    "    else:\n",
    "        overnight_flight.append(0)\n",
    "\n",
    "data[\"Overnight_Flight\"] = overnight_flight"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "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>Airline</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "      <th>Expensive_Day</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Overnight_Flight</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>22:20</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>17</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>01:10</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>05:50</td>\n",
       "      <td>7h 25m</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>55</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>13:15</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>09:25</td>\n",
       "      <td>19h</td>\n",
       "      <td>2 stops</td>\n",
       "      <td>94</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>04:25</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>18:05</td>\n",
       "      <td>5h 25m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>23:30</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>16:50</td>\n",
       "      <td>4h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>-6</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>21:35</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline  Source  Destination Dep_Time Duration Total_Stops  No_of_days  \\\n",
       "0        3       0            5    22:20   2h 50m    non-stop          17   \n",
       "1        1       3            0    05:50   7h 25m     2 stops          55   \n",
       "2        4       2            1    09:25      19h     2 stops          94   \n",
       "3        3       3            0    18:05   5h 25m      1 stop          66   \n",
       "4        3       0            5    16:50   4h 45m      1 stop          -6   \n",
       "\n",
       "   Weekday  Expensive_Day Arrival_Time  Overnight_Flight  \n",
       "0        6              0        01:10                 1  \n",
       "1        2              0        13:15                 0  \n",
       "2        6              0        04:25                 1  \n",
       "3        6              0        23:30                 0  \n",
       "4        4              1        21:35                 0  "
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = data.drop([\"Date_of_Journey\"],axis=1)\n",
    "data.head()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[170,\n",
       " 445,\n",
       " 1140,\n",
       " 325,\n",
       " 285,\n",
       " 145,\n",
       " 930,\n",
       " 1265,\n",
       " 1530,\n",
       " 470,\n",
       " 795,\n",
       " 155,\n",
       " 135,\n",
       " 730,\n",
       " 155,\n",
       " 1595,\n",
       " 270,\n",
       " 1355,\n",
       " 1380,\n",
       " 1235,\n",
       " 310,\n",
       " 920,\n",
       " 170,\n",
       " 175,\n",
       " 800,\n",
       " 910,\n",
       " 345,\n",
       " 355,\n",
       " 170,\n",
       " 135,\n",
       " 135,\n",
       " 805,\n",
       " 170,\n",
       " 1320,\n",
       " 330,\n",
       " 625,\n",
       " 315,\n",
       " 150,\n",
       " 375,\n",
       " 715,\n",
       " 665,\n",
       " 510,\n",
       " 1325,\n",
       " 165,\n",
       " 720,\n",
       " 170,\n",
       " 170,\n",
       " 135,\n",
       " 965,\n",
       " 1195,\n",
       " 195,\n",
       " 1520,\n",
       " 170,\n",
       " 180,\n",
       " 170,\n",
       " 975,\n",
       " 905,\n",
       " 135,\n",
       " 390,\n",
       " 1505,\n",
       " 745,\n",
       " 1640,\n",
       " 615,\n",
       " 630,\n",
       " 135,\n",
       " 625,\n",
       " 170,\n",
       " 90,\n",
       " 800,\n",
       " 135,\n",
       " 85,\n",
       " 1590,\n",
       " 440,\n",
       " 810,\n",
       " 300,\n",
       " 165,\n",
       " 170,\n",
       " 90,\n",
       " 1145,\n",
       " 135,\n",
       " 890,\n",
       " 160,\n",
       " 1330,\n",
       " 575,\n",
       " 600,\n",
       " 1280,\n",
       " 325,\n",
       " 1125,\n",
       " 740,\n",
       " 1080,\n",
       " 555,\n",
       " 665,\n",
       " 1050,\n",
       " 800,\n",
       " 155,\n",
       " 145,\n",
       " 160,\n",
       " 180,\n",
       " 85,\n",
       " 995,\n",
       " 740,\n",
       " 735,\n",
       " 450,\n",
       " 1440,\n",
       " 165,\n",
       " 535,\n",
       " 430,\n",
       " 870,\n",
       " 1820,\n",
       " 1195,\n",
       " 900,\n",
       " 175,\n",
       " 765,\n",
       " 610,\n",
       " 355,\n",
       " 925,\n",
       " 965,\n",
       " 610,\n",
       " 160,\n",
       " 715,\n",
       " 160,\n",
       " 85,\n",
       " 845,\n",
       " 1215,\n",
       " 145,\n",
       " 1390,\n",
       " 1090,\n",
       " 960,\n",
       " 170,\n",
       " 140,\n",
       " 480,\n",
       " 1015,\n",
       " 190,\n",
       " 165,\n",
       " 840,\n",
       " 1430,\n",
       " 1300,\n",
       " 1275,\n",
       " 165,\n",
       " 650,\n",
       " 480,\n",
       " 495,\n",
       " 325,\n",
       " 515,\n",
       " 710,\n",
       " 1655,\n",
       " 505,\n",
       " 1255,\n",
       " 290,\n",
       " 490,\n",
       " 140,\n",
       " 1465,\n",
       " 170,\n",
       " 740,\n",
       " 735,\n",
       " 1415,\n",
       " 995,\n",
       " 170,\n",
       " 1545,\n",
       " 1570,\n",
       " 1730,\n",
       " 165,\n",
       " 1515,\n",
       " 560,\n",
       " 270,\n",
       " 925,\n",
       " 90,\n",
       " 160,\n",
       " 550,\n",
       " 165,\n",
       " 1355,\n",
       " 185,\n",
       " 690,\n",
       " 570,\n",
       " 730,\n",
       " 1055,\n",
       " 305,\n",
       " 1550,\n",
       " 1200,\n",
       " 780,\n",
       " 1105,\n",
       " 90,\n",
       " 570,\n",
       " 960,\n",
       " 140,\n",
       " 270,\n",
       " 345,\n",
       " 1450,\n",
       " 155,\n",
       " 165,\n",
       " 295,\n",
       " 1535,\n",
       " 380,\n",
       " 185,\n",
       " 1120,\n",
       " 135,\n",
       " 730,\n",
       " 1165,\n",
       " 170,\n",
       " 505,\n",
       " 555,\n",
       " 170,\n",
       " 195,\n",
       " 170,\n",
       " 740,\n",
       " 175,\n",
       " 575,\n",
       " 155,\n",
       " 570,\n",
       " 1760,\n",
       " 175,\n",
       " 315,\n",
       " 545,\n",
       " 645,\n",
       " 295,\n",
       " 90,\n",
       " 700,\n",
       " 165,\n",
       " 175,\n",
       " 600,\n",
       " 165,\n",
       " 615,\n",
       " 85,\n",
       " 555,\n",
       " 1375,\n",
       " 1730,\n",
       " 2245,\n",
       " 170,\n",
       " 90,\n",
       " 90,\n",
       " 1540,\n",
       " 155,\n",
       " 375,\n",
       " 1050,\n",
       " 345,\n",
       " 155,\n",
       " 1530,\n",
       " 835,\n",
       " 175,\n",
       " 615,\n",
       " 520,\n",
       " 330,\n",
       " 720,\n",
       " 135,\n",
       " 135,\n",
       " 170,\n",
       " 550,\n",
       " 625,\n",
       " 140,\n",
       " 270,\n",
       " 190,\n",
       " 1410,\n",
       " 1640,\n",
       " 755,\n",
       " 1455,\n",
       " 80,\n",
       " 165,\n",
       " 180,\n",
       " 660,\n",
       " 810,\n",
       " 170,\n",
       " 175,\n",
       " 165,\n",
       " 675,\n",
       " 875,\n",
       " 835,\n",
       " 845,\n",
       " 905,\n",
       " 135,\n",
       " 900,\n",
       " 775,\n",
       " 540,\n",
       " 735,\n",
       " 460,\n",
       " 1640,\n",
       " 85,\n",
       " 80,\n",
       " 330,\n",
       " 705,\n",
       " 195,\n",
       " 1495,\n",
       " 1640,\n",
       " 175,\n",
       " 1025,\n",
       " 160,\n",
       " 480,\n",
       " 180,\n",
       " 1655,\n",
       " 450,\n",
       " 1375,\n",
       " 515,\n",
       " 1795,\n",
       " 1335,\n",
       " 80,\n",
       " 160,\n",
       " 180,\n",
       " 730,\n",
       " 155,\n",
       " 705,\n",
       " 155,\n",
       " 90,\n",
       " 160,\n",
       " 155,\n",
       " 1375,\n",
       " 880,\n",
       " 925,\n",
       " 435,\n",
       " 1210,\n",
       " 1245,\n",
       " 150,\n",
       " 1430,\n",
       " 1620,\n",
       " 330,\n",
       " 710,\n",
       " 135,\n",
       " 1470,\n",
       " 140,\n",
       " 390,\n",
       " 1225,\n",
       " 155,\n",
       " 175,\n",
       " 335,\n",
       " 150,\n",
       " 885,\n",
       " 740,\n",
       " 1530,\n",
       " 335,\n",
       " 705,\n",
       " 175,\n",
       " 340,\n",
       " 245,\n",
       " 480,\n",
       " 165,\n",
       " 160,\n",
       " 175,\n",
       " 955,\n",
       " 1390,\n",
       " 440,\n",
       " 170,\n",
       " 450,\n",
       " 180,\n",
       " 465,\n",
       " 1520,\n",
       " 625,\n",
       " 1820,\n",
       " 155,\n",
       " 540,\n",
       " 910,\n",
       " 1700,\n",
       " 180,\n",
       " 165,\n",
       " 295,\n",
       " 160,\n",
       " 1330,\n",
       " 610,\n",
       " 170,\n",
       " 260,\n",
       " 660,\n",
       " 180,\n",
       " 1355,\n",
       " 185,\n",
       " 160,\n",
       " 170,\n",
       " 1265,\n",
       " 710,\n",
       " 1535,\n",
       " 925,\n",
       " 145,\n",
       " 220,\n",
       " 175,\n",
       " 190,\n",
       " 710,\n",
       " 170,\n",
       " 170,\n",
       " 90,\n",
       " 690,\n",
       " 530,\n",
       " 1215,\n",
       " 1210,\n",
       " 965,\n",
       " 755,\n",
       " 1025,\n",
       " 180,\n",
       " 930,\n",
       " 690,\n",
       " 140,\n",
       " 1425,\n",
       " 90,\n",
       " 155,\n",
       " 1485,\n",
       " 1295,\n",
       " 330,\n",
       " 735,\n",
       " 485,\n",
       " 300,\n",
       " 170,\n",
       " 385,\n",
       " 140,\n",
       " 165,\n",
       " 765,\n",
       " 950,\n",
       " 1620,\n",
       " 1585,\n",
       " 175,\n",
       " 1490,\n",
       " 520,\n",
       " 1545,\n",
       " 170,\n",
       " 375,\n",
       " 630,\n",
       " 1560,\n",
       " 1025,\n",
       " 1280,\n",
       " 335,\n",
       " 170,\n",
       " 1550,\n",
       " 730,\n",
       " 1025,\n",
       " 510,\n",
       " 1385,\n",
       " 560,\n",
       " 475,\n",
       " 90,\n",
       " 170,\n",
       " 135,\n",
       " 600,\n",
       " 155,\n",
       " 465,\n",
       " 175,\n",
       " 1580,\n",
       " 160,\n",
       " 660,\n",
       " 1165,\n",
       " 175,\n",
       " 165,\n",
       " 1395,\n",
       " 540,\n",
       " 190,\n",
       " 385,\n",
       " 520,\n",
       " 170,\n",
       " 740,\n",
       " 320,\n",
       " 240,\n",
       " 585,\n",
       " 175,\n",
       " 1580,\n",
       " 135,\n",
       " 585,\n",
       " 1495,\n",
       " 715,\n",
       " 500,\n",
       " 1045,\n",
       " 170,\n",
       " 425,\n",
       " 1275,\n",
       " 300,\n",
       " 145,\n",
       " 315,\n",
       " 910,\n",
       " 705,\n",
       " 180,\n",
       " 295,\n",
       " 165,\n",
       " 150,\n",
       " 585,\n",
       " 160,\n",
       " 390,\n",
       " 330,\n",
       " 160,\n",
       " 2045,\n",
       " 925,\n",
       " 570,\n",
       " 365,\n",
       " 160,\n",
       " 350,\n",
       " 165,\n",
       " 1640,\n",
       " 180,\n",
       " 315,\n",
       " 960,\n",
       " 420,\n",
       " 520,\n",
       " 540,\n",
       " 140,\n",
       " 140,\n",
       " 380,\n",
       " 265,\n",
       " 140,\n",
       " 665,\n",
       " 1015,\n",
       " 825,\n",
       " 1545,\n",
       " 835,\n",
       " 315,\n",
       " 1155,\n",
       " 450,\n",
       " 540,\n",
       " 1440,\n",
       " 1090,\n",
       " 175,\n",
       " 90,\n",
       " 1280,\n",
       " 375,\n",
       " 870,\n",
       " 80,\n",
       " 775,\n",
       " 145,\n",
       " 1350,\n",
       " 1375,\n",
       " 325,\n",
       " 85,\n",
       " 435,\n",
       " 740,\n",
       " 1210,\n",
       " 1275,\n",
       " 135,\n",
       " 630,\n",
       " 845,\n",
       " 170,\n",
       " 985,\n",
       " 315,\n",
       " 830,\n",
       " 135,\n",
       " 435,\n",
       " 1625,\n",
       " 1655,\n",
       " 300,\n",
       " 170,\n",
       " 170,\n",
       " 920,\n",
       " 320,\n",
       " 610,\n",
       " 1655,\n",
       " 575,\n",
       " 1690,\n",
       " 840,\n",
       " 155,\n",
       " 175,\n",
       " 155,\n",
       " 280,\n",
       " 150,\n",
       " 675,\n",
       " 475,\n",
       " 530,\n",
       " 85,\n",
       " 940,\n",
       " 155,\n",
       " 175,\n",
       " 805,\n",
       " 275,\n",
       " 1110,\n",
       " 135,\n",
       " 2295,\n",
       " 185,\n",
       " 1550,\n",
       " 190,\n",
       " 170,\n",
       " 690,\n",
       " 395,\n",
       " 1655,\n",
       " 155,\n",
       " 450,\n",
       " 195,\n",
       " 575,\n",
       " 750,\n",
       " 680,\n",
       " 180,\n",
       " 155,\n",
       " 90,\n",
       " 1570,\n",
       " 295,\n",
       " 165,\n",
       " 520,\n",
       " 425,\n",
       " 165,\n",
       " 1490,\n",
       " 140,\n",
       " 660,\n",
       " 540,\n",
       " 175,\n",
       " 185,\n",
       " 155,\n",
       " 780,\n",
       " 85,\n",
       " 455,\n",
       " 160,\n",
       " 775,\n",
       " 1760,\n",
       " 300,\n",
       " 520,\n",
       " 180,\n",
       " 180,\n",
       " 165,\n",
       " 730,\n",
       " 140,\n",
       " 475,\n",
       " 1775,\n",
       " 1615,\n",
       " 765,\n",
       " 455,\n",
       " 150,\n",
       " 705,\n",
       " 1420,\n",
       " 830,\n",
       " 770,\n",
       " 590,\n",
       " 150,\n",
       " 1315,\n",
       " 1570,\n",
       " 175,\n",
       " 1215,\n",
       " 1210,\n",
       " 655,\n",
       " 150,\n",
       " 1270,\n",
       " 155,\n",
       " 1240,\n",
       " 170,\n",
       " 390,\n",
       " 1800,\n",
       " 790,\n",
       " 175,\n",
       " 1080,\n",
       " 480,\n",
       " 525,\n",
       " 750,\n",
       " 800,\n",
       " 370,\n",
       " 450,\n",
       " 1155,\n",
       " 155,\n",
       " 720,\n",
       " 180,\n",
       " 170,\n",
       " 85,\n",
       " 1385,\n",
       " 170,\n",
       " 1325,\n",
       " 135,\n",
       " 345,\n",
       " 170,\n",
       " 570,\n",
       " 535,\n",
       " 1065,\n",
       " 170,\n",
       " 1305,\n",
       " 755,\n",
       " 90,\n",
       " 135,\n",
       " 235,\n",
       " 165,\n",
       " 450,\n",
       " 1040,\n",
       " 90,\n",
       " 90,\n",
       " 1830,\n",
       " 300,\n",
       " 675,\n",
       " 170,\n",
       " 1145,\n",
       " 395,\n",
       " 450,\n",
       " 135,\n",
       " 165,\n",
       " 765,\n",
       " 170,\n",
       " 1595,\n",
       " 775,\n",
       " 90,\n",
       " 520,\n",
       " 840,\n",
       " 1410,\n",
       " 1795,\n",
       " 1330,\n",
       " 290,\n",
       " 165,\n",
       " 810,\n",
       " 800,\n",
       " 140,\n",
       " 715,\n",
       " 310,\n",
       " 705,\n",
       " 830,\n",
       " 180,\n",
       " 370,\n",
       " 140,\n",
       " 645,\n",
       " 135,\n",
       " 175,\n",
       " 90,\n",
       " 455,\n",
       " 1595,\n",
       " 285,\n",
       " 1285,\n",
       " 515,\n",
       " 510,\n",
       " 135,\n",
       " 760,\n",
       " 505,\n",
       " 1540,\n",
       " 165,\n",
       " 1760,\n",
       " 160,\n",
       " 170,\n",
       " 140,\n",
       " 435,\n",
       " 1215,\n",
       " 705,\n",
       " 155,\n",
       " 810,\n",
       " 90,\n",
       " 1475,\n",
       " 1235,\n",
       " 170,\n",
       " 275,\n",
       " 140,\n",
       " 705,\n",
       " 90,\n",
       " 770,\n",
       " 1470,\n",
       " 690,\n",
       " 240,\n",
       " 660,\n",
       " 325,\n",
       " 155,\n",
       " 310,\n",
       " 610,\n",
       " 1615,\n",
       " 435,\n",
       " 490,\n",
       " 175,\n",
       " 315,\n",
       " 140,\n",
       " 85,\n",
       " 290,\n",
       " 140,\n",
       " 140,\n",
       " 795,\n",
       " 1150,\n",
       " 1360,\n",
       " 890,\n",
       " 910,\n",
       " 425,\n",
       " 85,\n",
       " 175,\n",
       " 480,\n",
       " 165,\n",
       " 895,\n",
       " 90,\n",
       " 975,\n",
       " 170,\n",
       " 300,\n",
       " 1335,\n",
       " 330,\n",
       " 515,\n",
       " 925,\n",
       " 1260,\n",
       " 810,\n",
       " 170,\n",
       " 405,\n",
       " 90,\n",
       " 520,\n",
       " 90,\n",
       " 170,\n",
       " 155,\n",
       " 965,\n",
       " 1430,\n",
       " 1625,\n",
       " 975,\n",
       " 925,\n",
       " 170,\n",
       " 515,\n",
       " 830,\n",
       " 160,\n",
       " 145,\n",
       " 190,\n",
       " 175,\n",
       " 1720,\n",
       " 170,\n",
       " 155,\n",
       " 580,\n",
       " 1000,\n",
       " 380,\n",
       " 1620,\n",
       " 875,\n",
       " 730,\n",
       " 455,\n",
       " 140,\n",
       " 1150,\n",
       " 1200,\n",
       " 180,\n",
       " 310,\n",
       " 160,\n",
       " 1775,\n",
       " 295,\n",
       " 180,\n",
       " 750,\n",
       " 755,\n",
       " 1580,\n",
       " 795,\n",
       " 170,\n",
       " 765,\n",
       " 980,\n",
       " 160,\n",
       " 495,\n",
       " 720,\n",
       " 170,\n",
       " 660,\n",
       " 140,\n",
       " 830,\n",
       " 490,\n",
       " 730,\n",
       " 160,\n",
       " 165,\n",
       " 170,\n",
       " 1440,\n",
       " 180,\n",
       " 755,\n",
       " 1550,\n",
       " 175,\n",
       " 1025,\n",
       " 1760,\n",
       " 535,\n",
       " 185,\n",
       " 540,\n",
       " 625,\n",
       " 1005,\n",
       " 165,\n",
       " 1580,\n",
       " 795,\n",
       " 630,\n",
       " 455,\n",
       " 660,\n",
       " 170,\n",
       " 1545,\n",
       " 385,\n",
       " 85,\n",
       " 1210,\n",
       " 715,\n",
       " 170,\n",
       " 300,\n",
       " 1620,\n",
       " 170,\n",
       " 75,\n",
       " 615,\n",
       " 1490,\n",
       " 170,\n",
       " 1200,\n",
       " 330,\n",
       " 165,\n",
       " 415,\n",
       " 165,\n",
       " 315,\n",
       " 375,\n",
       " 170,\n",
       " 1165,\n",
       " 175,\n",
       " 685,\n",
       " 845,\n",
       " 165,\n",
       " 195,\n",
       " 860,\n",
       " 750,\n",
       " 320,\n",
       " 1580,\n",
       " 185,\n",
       " 90,\n",
       " 425,\n",
       " 370,\n",
       " 480,\n",
       " 725,\n",
       " 775,\n",
       " 575,\n",
       " 910,\n",
       " 700,\n",
       " 175,\n",
       " 270,\n",
       " 160,\n",
       " 150,\n",
       " 1450,\n",
       " 1700,\n",
       " 85,\n",
       " 610,\n",
       " 690,\n",
       " 170,\n",
       " 825,\n",
       " 385,\n",
       " 775,\n",
       " 265,\n",
       " 810,\n",
       " 1445,\n",
       " 965,\n",
       " 500,\n",
       " 220,\n",
       " 570,\n",
       " 1695,\n",
       " 800,\n",
       " 1070,\n",
       " 1220,\n",
       " 1685,\n",
       " 925,\n",
       " 725,\n",
       " 620,\n",
       " 745,\n",
       " 455,\n",
       " 170,\n",
       " 275,\n",
       " 675,\n",
       " 155,\n",
       " 1560,\n",
       " 145,\n",
       " 155,\n",
       " 855,\n",
       " 155,\n",
       " 1325,\n",
       " 395,\n",
       " 510,\n",
       " 1125,\n",
       " 2115,\n",
       " 135,\n",
       " 150,\n",
       " 140,\n",
       " 150,\n",
       " 455,\n",
       " 2135,\n",
       " 1590,\n",
       " 690,\n",
       " 1600,\n",
       " 180,\n",
       " 90,\n",
       " 175,\n",
       " 855,\n",
       " 1015,\n",
       " 740,\n",
       " 1775,\n",
       " 1220,\n",
       " 505,\n",
       " 830,\n",
       " 600,\n",
       " 1165,\n",
       " 1680,\n",
       " 170,\n",
       " 1040,\n",
       " 165,\n",
       " 175,\n",
       " 805,\n",
       " 775,\n",
       " 585,\n",
       " 1285,\n",
       " 940,\n",
       " 140,\n",
       " 865,\n",
       " 170,\n",
       " 395,\n",
       " 1330,\n",
       " 865,\n",
       " 150,\n",
       " 1415,\n",
       " 785,\n",
       " 90,\n",
       " 305,\n",
       " 80,\n",
       " 1375,\n",
       " 90,\n",
       " 155,\n",
       " 2240,\n",
       " 170,\n",
       " 170,\n",
       " 2170,\n",
       " 555,\n",
       " 85,\n",
       " 305,\n",
       " 135,\n",
       " 910,\n",
       " 1555,\n",
       " 340,\n",
       " 450,\n",
       " 1065,\n",
       " 140,\n",
       " 495,\n",
       " 220,\n",
       " 355,\n",
       " 150,\n",
       " 1050,\n",
       " 830,\n",
       " 135,\n",
       " 1295,\n",
       " 570,\n",
       " 520,\n",
       " 625,\n",
       " 2105,\n",
       " 855,\n",
       " 180,\n",
       " 1550,\n",
       " 150,\n",
       " 520,\n",
       " 600,\n",
       " 690,\n",
       " 450,\n",
       " ...]"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import re\n",
    "new_dur = []\n",
    "duration = data[\"Duration\"]\n",
    "overnight = data[\"Overnight_Flight\"]\n",
    "\n",
    "for string in duration:\n",
    "    s = str(string)\n",
    "    a = []\n",
    "    a = re.findall(r'\\d+', s)\n",
    "    if len(a)==2:\n",
    "        new_dur.append((int(a[0])*60)+int(a[1]))\n",
    "    else:\n",
    "        new_dur.append(int(a[0])*60)\n",
    "\n",
    "new_dur\n",
    "\n",
    "# for i,j in zip(duration,overnight):\n",
    "#     if j==1:\n",
    "#         hours,minutes = i.split(\" \")\n",
    "# #         hours = hours[:-1]\n",
    "#         minutes = minutes[:-1]\n",
    "#         new_dur.append((hours*60)+minutes)\n",
    "#     else:\n",
    "#         hours = i.split(\" \")\n",
    "#         new_dur.append((hours[:-1])*60)\n",
    "# new_dur"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "stops = data[\"Total_Stops\"]\n",
    "new_Stops = []\n",
    "for i in stops:\n",
    "    if i==\"non-stop\":\n",
    "        new_Stops.append(0)\n",
    "    elif i==\"1 stop\":\n",
    "        new_Stops.append(1)\n",
    "    elif i==\"2 stops\":\n",
    "        new_Stops.append(2)\n",
    "    else:\n",
    "        new_Stops.append(3)\n",
    "\n",
    "data = data.drop([\"Dep_Time\",\"Arrival_Time\",\"Duration\",\"Total_Stops\"],axis=1)\n",
    "data[\"Duration\"] = new_dur\n",
    "data[\"Total_Stops\"] = new_Stops"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Airline</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "      <th>Overnight_Flight</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>17</td>\n",
       "      <td>6</td>\n",
       "      <td>1</td>\n",
       "      <td>170</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>55</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>445</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>94</td>\n",
       "      <td>6</td>\n",
       "      <td>1</td>\n",
       "      <td>1140</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>325</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>-6</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>285</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline  Source  Destination  No_of_days  Weekday  Overnight_Flight  \\\n",
       "0        3       0            5          17        6                 1   \n",
       "1        1       3            0          55        2                 0   \n",
       "2        4       2            1          94        6                 1   \n",
       "3        3       3            0          66        6                 0   \n",
       "4        3       0            5          -6        4                 0   \n",
       "\n",
       "   Duration  Total_Stops  \n",
       "0       170            0  \n",
       "1       445            2  \n",
       "2      1140            2  \n",
       "3       325            1  \n",
       "4       285            1  "
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = data.drop([\"Expensive_Day\"],axis=1)\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [],
   "source": [
    "# data.to_csv(\"training_Data.csv\")\n",
    "# label.to_csv(\"training_labels.csv\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "DecisionTreeRegressor(criterion='mse', max_depth=None, max_features=None,\n",
       "           max_leaf_nodes=None, min_impurity_decrease=0.0,\n",
       "           min_impurity_split=None, min_samples_leaf=1,\n",
       "           min_samples_split=2, min_weight_fraction_leaf=0.0,\n",
       "           presort=False, random_state=0, splitter='best')"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from sklearn.tree import DecisionTreeRegressor\n",
    "\n",
    "regressor = DecisionTreeRegressor(random_state=0)\n",
    "regressor.fit(data,label)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0     3897\n",
       "1     7662\n",
       "2    13882\n",
       "3     6218\n",
       "4    13302\n",
       "Name: Price, dtype: int64"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "label.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Checking the model out on test_Data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>Additional_Info</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>6/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>17:30</td>\n",
       "      <td>04:25 07 Jun</td>\n",
       "      <td>10h 55m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>No info</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → MAA → BLR</td>\n",
       "      <td>06:20</td>\n",
       "      <td>10:20</td>\n",
       "      <td>4h</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>No info</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>19:15</td>\n",
       "      <td>19:00 22 May</td>\n",
       "      <td>23h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>In-flight meal not included</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Multiple carriers</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>08:00</td>\n",
       "      <td>21:00</td>\n",
       "      <td>13h</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>No info</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Air Asia</td>\n",
       "      <td>24/06/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>BLR → DEL</td>\n",
       "      <td>23:55</td>\n",
       "      <td>02:45 25 Jun</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>No info</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Airline Date_of_Journey    Source Destination            Route  \\\n",
       "0        Jet Airways       6/06/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "1             IndiGo      12/05/2019   Kolkata    Banglore  CCU → MAA → BLR   \n",
       "2        Jet Airways      21/05/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "3  Multiple carriers      21/05/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "4           Air Asia      24/06/2019  Banglore       Delhi        BLR → DEL   \n",
       "\n",
       "  Dep_Time  Arrival_Time Duration Total_Stops              Additional_Info  \n",
       "0    17:30  04:25 07 Jun  10h 55m      1 stop                      No info  \n",
       "1    06:20         10:20       4h      1 stop                      No info  \n",
       "2    19:15  19:00 22 May  23h 45m      1 stop  In-flight meal not included  \n",
       "3    08:00         21:00      13h      1 stop                      No info  \n",
       "4    23:55  02:45 25 Jun   2h 50m    non-stop                      No info  "
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_data = pd.read_excel(\"Flight_Ticket_Participant_Datasets/Test_set.xlsx\")\n",
    "test_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>6/06/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>17:30</td>\n",
       "      <td>04:25 07 Jun</td>\n",
       "      <td>10h 55m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>IndiGo</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>Kolkata</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>CCU → MAA → BLR</td>\n",
       "      <td>06:20</td>\n",
       "      <td>10:20</td>\n",
       "      <td>4h</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jet Airways</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>19:15</td>\n",
       "      <td>19:00 22 May</td>\n",
       "      <td>23h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Multiple carriers</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>Cochin</td>\n",
       "      <td>DEL → BOM → COK</td>\n",
       "      <td>08:00</td>\n",
       "      <td>21:00</td>\n",
       "      <td>13h</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Air Asia</td>\n",
       "      <td>24/06/2019</td>\n",
       "      <td>Banglore</td>\n",
       "      <td>Delhi</td>\n",
       "      <td>BLR → DEL</td>\n",
       "      <td>23:55</td>\n",
       "      <td>02:45 25 Jun</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Airline Date_of_Journey    Source Destination            Route  \\\n",
       "0        Jet Airways       6/06/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "1             IndiGo      12/05/2019   Kolkata    Banglore  CCU → MAA → BLR   \n",
       "2        Jet Airways      21/05/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "3  Multiple carriers      21/05/2019     Delhi      Cochin  DEL → BOM → COK   \n",
       "4           Air Asia      24/06/2019  Banglore       Delhi        BLR → DEL   \n",
       "\n",
       "  Dep_Time  Arrival_Time Duration Total_Stops  \n",
       "0    17:30  04:25 07 Jun  10h 55m      1 stop  \n",
       "1    06:20         10:20       4h      1 stop  \n",
       "2    19:15  19:00 22 May  23h 45m      1 stop  \n",
       "3    08:00         21:00      13h      1 stop  \n",
       "4    23:55  02:45 25 Jun   2h 50m    non-stop  "
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_data = test_data.drop([\"Additional_Info\"],axis=1)\n",
    "test_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>4</td>\n",
       "      <td>6/06/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>17:30</td>\n",
       "      <td>04:25 07 Jun</td>\n",
       "      <td>10h 55m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>65</td>\n",
       "      <td>06:20</td>\n",
       "      <td>10:20</td>\n",
       "      <td>4h</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>19:15</td>\n",
       "      <td>19:00 22 May</td>\n",
       "      <td>23h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>6</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>08:00</td>\n",
       "      <td>21:00</td>\n",
       "      <td>13h</td>\n",
       "      <td>1 stop</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>24/06/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>16</td>\n",
       "      <td>23:55</td>\n",
       "      <td>02:45 25 Jun</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline Date_of_Journey  Source  Destination  Route Dep_Time  Arrival_Time  \\\n",
       "0        4       6/06/2019       2            1     76    17:30  04:25 07 Jun   \n",
       "1        3      12/05/2019       3            0     65    06:20         10:20   \n",
       "2        4      21/05/2019       2            1     76    19:15  19:00 22 May   \n",
       "3        6      21/05/2019       2            1     76    08:00         21:00   \n",
       "4        0      24/06/2019       0            2     16    23:55  02:45 25 Jun   \n",
       "\n",
       "  Duration Total_Stops  \n",
       "0  10h 55m      1 stop  \n",
       "1       4h      1 stop  \n",
       "2  23h 45m      1 stop  \n",
       "3      13h      1 stop  \n",
       "4   2h 50m    non-stop  "
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from sklearn import preprocessing\n",
    "\n",
    "le = preprocessing.LabelEncoder()\n",
    "\n",
    "le.fit(test_data[\"Airline\"])\n",
    "test_data[\"Airline\"] = le.transform(test_data[\"Airline\"])\n",
    "\n",
    "le.fit(test_data[\"Route\"])\n",
    "test_data[\"Route\"] = le.transform(test_data[\"Route\"])\n",
    "\n",
    "le.fit(test_data[\"Source\"])\n",
    "test_data[\"Source\"] = le.transform(test_data[\"Source\"])\n",
    "\n",
    "le.fit(test_data[\"Destination\"])\n",
    "test_data[\"Destination\"] = le.transform(test_data[\"Destination\"])\n",
    "\n",
    "test_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "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>Airline</th>\n",
       "      <th>Date_of_Journey</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>Route</th>\n",
       "      <th>Dep_Time</th>\n",
       "      <th>Arrival_Time</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "      <th>Overnight_Flight</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>4</td>\n",
       "      <td>6/06/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>17:30</td>\n",
       "      <td>04:25 07 Jun</td>\n",
       "      <td>10h 55m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3</td>\n",
       "      <td>12/05/2019</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>65</td>\n",
       "      <td>06:20</td>\n",
       "      <td>10:20</td>\n",
       "      <td>4h</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>19:15</td>\n",
       "      <td>19:00 22 May</td>\n",
       "      <td>23h 45m</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>6</td>\n",
       "      <td>21/05/2019</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>76</td>\n",
       "      <td>08:00</td>\n",
       "      <td>21:00</td>\n",
       "      <td>13h</td>\n",
       "      <td>1 stop</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>24/06/2019</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>16</td>\n",
       "      <td>23:55</td>\n",
       "      <td>02:45 25 Jun</td>\n",
       "      <td>2h 50m</td>\n",
       "      <td>non-stop</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline Date_of_Journey  Source  Destination  Route Dep_Time  Arrival_Time  \\\n",
       "0        4       6/06/2019       2            1     76    17:30  04:25 07 Jun   \n",
       "1        3      12/05/2019       3            0     65    06:20         10:20   \n",
       "2        4      21/05/2019       2            1     76    19:15  19:00 22 May   \n",
       "3        6      21/05/2019       2            1     76    08:00         21:00   \n",
       "4        0      24/06/2019       0            2     16    23:55  02:45 25 Jun   \n",
       "\n",
       "  Duration Total_Stops  Overnight_Flight  \n",
       "0  10h 55m      1 stop                 1  \n",
       "1       4h      1 stop                 0  \n",
       "2  23h 45m      1 stop                 1  \n",
       "3      13h      1 stop                 0  \n",
       "4   2h 50m    non-stop                 1  "
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "overnight_flight = []\n",
    "dep = test_data[\"Dep_Time\"]\n",
    "arr = test_data[\"Arrival_Time\"]\n",
    "for i,j in zip(dep,arr):\n",
    "    if i>j:\n",
    "        overnight_flight.append(1)\n",
    "    else:\n",
    "        overnight_flight.append(0)\n",
    "\n",
    "test_data[\"Overnight_Flight\"] = overnight_flight\n",
    "test_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[655,\n",
       " 240,\n",
       " 1425,\n",
       " 780,\n",
       " 170,\n",
       " 1100,\n",
       " 905,\n",
       " 315,\n",
       " 165,\n",
       " 365,\n",
       " 1735,\n",
       " 560,\n",
       " 615,\n",
       " 280,\n",
       " 1040,\n",
       " 650,\n",
       " 555,\n",
       " 735,\n",
       " 550,\n",
       " 560,\n",
       " 890,\n",
       " 845,\n",
       " 695,\n",
       " 810,\n",
       " 170,\n",
       " 745,\n",
       " 150,\n",
       " 1530,\n",
       " 1165,\n",
       " 420,\n",
       " 150,\n",
       " 700,\n",
       " 425,\n",
       " 180,\n",
       " 85,\n",
       " 160,\n",
       " 85,\n",
       " 365,\n",
       " 1025,\n",
       " 640,\n",
       " 150,\n",
       " 750,\n",
       " 90,\n",
       " 1420,\n",
       " 315,\n",
       " 140,\n",
       " 1415,\n",
       " 90,\n",
       " 1660,\n",
       " 810,\n",
       " 1700,\n",
       " 345,\n",
       " 1585,\n",
       " 340,\n",
       " 135,\n",
       " 175,\n",
       " 345,\n",
       " 450,\n",
       " 320,\n",
       " 1550,\n",
       " 795,\n",
       " 1580,\n",
       " 345,\n",
       " 340,\n",
       " 160,\n",
       " 170,\n",
       " 325,\n",
       " 525,\n",
       " 180,\n",
       " 440,\n",
       " 90,\n",
       " 1270,\n",
       " 960,\n",
       " 470,\n",
       " 90,\n",
       " 180,\n",
       " 170,\n",
       " 1165,\n",
       " 175,\n",
       " 845,\n",
       " 170,\n",
       " 965,\n",
       " 355,\n",
       " 150,\n",
       " 615,\n",
       " 830,\n",
       " 135,\n",
       " 715,\n",
       " 150,\n",
       " 800,\n",
       " 1090,\n",
       " 1690,\n",
       " 300,\n",
       " 1395,\n",
       " 75,\n",
       " 80,\n",
       " 805,\n",
       " 375,\n",
       " 80,\n",
       " 610,\n",
       " 275,\n",
       " 660,\n",
       " 470,\n",
       " 740,\n",
       " 150,\n",
       " 910,\n",
       " 90,\n",
       " 1140,\n",
       " 825,\n",
       " 740,\n",
       " 590,\n",
       " 845,\n",
       " 155,\n",
       " 280,\n",
       " 435,\n",
       " 1380,\n",
       " 625,\n",
       " 305,\n",
       " 1365,\n",
       " 520,\n",
       " 1660,\n",
       " 690,\n",
       " 155,\n",
       " 165,\n",
       " 315,\n",
       " 940,\n",
       " 150,\n",
       " 155,\n",
       " 560,\n",
       " 170,\n",
       " 520,\n",
       " 320,\n",
       " 345,\n",
       " 150,\n",
       " 165,\n",
       " 475,\n",
       " 1165,\n",
       " 170,\n",
       " 165,\n",
       " 770,\n",
       " 165,\n",
       " 1525,\n",
       " 620,\n",
       " 510,\n",
       " 180,\n",
       " 765,\n",
       " 180,\n",
       " 150,\n",
       " 730,\n",
       " 1350,\n",
       " 475,\n",
       " 465,\n",
       " 420,\n",
       " 175,\n",
       " 150,\n",
       " 175,\n",
       " 160,\n",
       " 170,\n",
       " 1315,\n",
       " 90,\n",
       " 180,\n",
       " 390,\n",
       " 180,\n",
       " 155,\n",
       " 170,\n",
       " 165,\n",
       " 1580,\n",
       " 140,\n",
       " 520,\n",
       " 145,\n",
       " 1380,\n",
       " 560,\n",
       " 175,\n",
       " 165,\n",
       " 1545,\n",
       " 150,\n",
       " 315,\n",
       " 2000,\n",
       " 940,\n",
       " 845,\n",
       " 745,\n",
       " 90,\n",
       " 140,\n",
       " 685,\n",
       " 615,\n",
       " 915,\n",
       " 870,\n",
       " 480,\n",
       " 470,\n",
       " 140,\n",
       " 780,\n",
       " 750,\n",
       " 90,\n",
       " 640,\n",
       " 1275,\n",
       " 145,\n",
       " 1610,\n",
       " 1010,\n",
       " 165,\n",
       " 545,\n",
       " 610,\n",
       " 1005,\n",
       " 1350,\n",
       " 135,\n",
       " 675,\n",
       " 160,\n",
       " 1165,\n",
       " 1375,\n",
       " 695,\n",
       " 1065,\n",
       " 180,\n",
       " 1390,\n",
       " 1520,\n",
       " 320,\n",
       " 465,\n",
       " 1430,\n",
       " 365,\n",
       " 865,\n",
       " 1025,\n",
       " 1490,\n",
       " 170,\n",
       " 630,\n",
       " 435,\n",
       " 140,\n",
       " 150,\n",
       " 1180,\n",
       " 155,\n",
       " 90,\n",
       " 380,\n",
       " 1005,\n",
       " 1100,\n",
       " 1640,\n",
       " 85,\n",
       " 565,\n",
       " 175,\n",
       " 310,\n",
       " 460,\n",
       " 165,\n",
       " 395,\n",
       " 965,\n",
       " 1730,\n",
       " 780,\n",
       " 875,\n",
       " 180,\n",
       " 625,\n",
       " 1195,\n",
       " 810,\n",
       " 160,\n",
       " 660,\n",
       " 845,\n",
       " 1380,\n",
       " 2280,\n",
       " 375,\n",
       " 285,\n",
       " 385,\n",
       " 1300,\n",
       " 545,\n",
       " 185,\n",
       " 1390,\n",
       " 295,\n",
       " 970,\n",
       " 90,\n",
       " 180,\n",
       " 405,\n",
       " 165,\n",
       " 170,\n",
       " 810,\n",
       " 180,\n",
       " 425,\n",
       " 705,\n",
       " 360,\n",
       " 1345,\n",
       " 170,\n",
       " 870,\n",
       " 385,\n",
       " 160,\n",
       " 910,\n",
       " 1150,\n",
       " 345,\n",
       " 175,\n",
       " 625,\n",
       " 550,\n",
       " 765,\n",
       " 1440,\n",
       " 880,\n",
       " 160,\n",
       " 175,\n",
       " 1350,\n",
       " 300,\n",
       " 940,\n",
       " 320,\n",
       " 1230,\n",
       " 1420,\n",
       " 975,\n",
       " 1375,\n",
       " 495,\n",
       " 505,\n",
       " 1385,\n",
       " 1170,\n",
       " 1580,\n",
       " 315,\n",
       " 170,\n",
       " 135,\n",
       " 750,\n",
       " 1510,\n",
       " 845,\n",
       " 1285,\n",
       " 90,\n",
       " 520,\n",
       " 290,\n",
       " 170,\n",
       " 625,\n",
       " 1105,\n",
       " 590,\n",
       " 85,\n",
       " 1180,\n",
       " 380,\n",
       " 590,\n",
       " 610,\n",
       " 245,\n",
       " 1425,\n",
       " 180,\n",
       " 905,\n",
       " 1005,\n",
       " 1000,\n",
       " 145,\n",
       " 1680,\n",
       " 1495,\n",
       " 625,\n",
       " 1105,\n",
       " 180,\n",
       " 170,\n",
       " 80,\n",
       " 170,\n",
       " 305,\n",
       " 180,\n",
       " 415,\n",
       " 1690,\n",
       " 500,\n",
       " 1560,\n",
       " 175,\n",
       " 765,\n",
       " 540,\n",
       " 80,\n",
       " 1255,\n",
       " 155,\n",
       " 1400,\n",
       " 290,\n",
       " 1655,\n",
       " 505,\n",
       " 1295,\n",
       " 930,\n",
       " 740,\n",
       " 840,\n",
       " 1720,\n",
       " 885,\n",
       " 670,\n",
       " 1350,\n",
       " 375,\n",
       " 555,\n",
       " 155,\n",
       " 275,\n",
       " 90,\n",
       " 780,\n",
       " 965,\n",
       " 705,\n",
       " 85,\n",
       " 745,\n",
       " 175,\n",
       " 1200,\n",
       " 175,\n",
       " 870,\n",
       " 160,\n",
       " 155,\n",
       " 1455,\n",
       " 750,\n",
       " 720,\n",
       " 1545,\n",
       " 325,\n",
       " 520,\n",
       " 1210,\n",
       " 675,\n",
       " 840,\n",
       " 165,\n",
       " 540,\n",
       " 870,\n",
       " 90,\n",
       " 155,\n",
       " 150,\n",
       " 800,\n",
       " 1430,\n",
       " 720,\n",
       " 170,\n",
       " 160,\n",
       " 1665,\n",
       " 915,\n",
       " 495,\n",
       " 175,\n",
       " 1235,\n",
       " 170,\n",
       " 665,\n",
       " 165,\n",
       " 870,\n",
       " 150,\n",
       " 460,\n",
       " 1375,\n",
       " 155,\n",
       " 755,\n",
       " 470,\n",
       " 735,\n",
       " 625,\n",
       " 140,\n",
       " 1500,\n",
       " 600,\n",
       " 1130,\n",
       " 170,\n",
       " 1580,\n",
       " 1125,\n",
       " 780,\n",
       " 485,\n",
       " 300,\n",
       " 265,\n",
       " 895,\n",
       " 905,\n",
       " 1325,\n",
       " 90,\n",
       " 180,\n",
       " 160,\n",
       " 625,\n",
       " 585,\n",
       " 1555,\n",
       " 1505,\n",
       " 90,\n",
       " 145,\n",
       " 155,\n",
       " 1200,\n",
       " 250,\n",
       " 655,\n",
       " 520,\n",
       " 520,\n",
       " 1010,\n",
       " 80,\n",
       " 575,\n",
       " 135,\n",
       " 705,\n",
       " 560,\n",
       " 1615,\n",
       " 155,\n",
       " 710,\n",
       " 555,\n",
       " 455,\n",
       " 1025,\n",
       " 710,\n",
       " 1350,\n",
       " 335,\n",
       " 545,\n",
       " 165,\n",
       " 345,\n",
       " 705,\n",
       " 330,\n",
       " 2135,\n",
       " 1035,\n",
       " 945,\n",
       " 765,\n",
       " 195,\n",
       " 1125,\n",
       " 1255,\n",
       " 195,\n",
       " 150,\n",
       " 985,\n",
       " 1350,\n",
       " 1495,\n",
       " 140,\n",
       " 730,\n",
       " 650,\n",
       " 625,\n",
       " 155,\n",
       " 160,\n",
       " 1655,\n",
       " 1540,\n",
       " 735,\n",
       " 690,\n",
       " 760,\n",
       " 355,\n",
       " 960,\n",
       " 460,\n",
       " 775,\n",
       " 635,\n",
       " 600,\n",
       " 505,\n",
       " 310,\n",
       " 1915,\n",
       " 450,\n",
       " 520,\n",
       " 140,\n",
       " 145,\n",
       " 1390,\n",
       " 1615,\n",
       " 170,\n",
       " 590,\n",
       " 320,\n",
       " 910,\n",
       " 480,\n",
       " 1245,\n",
       " 140,\n",
       " 1285,\n",
       " 620,\n",
       " 635,\n",
       " 150,\n",
       " 670,\n",
       " 2295,\n",
       " 360,\n",
       " 170,\n",
       " 90,\n",
       " 620,\n",
       " 985,\n",
       " 825,\n",
       " 380,\n",
       " 155,\n",
       " 140,\n",
       " 185,\n",
       " 780,\n",
       " 170,\n",
       " 360,\n",
       " 1375,\n",
       " 1485,\n",
       " 95,\n",
       " 1470,\n",
       " 165,\n",
       " 90,\n",
       " 455,\n",
       " 2070,\n",
       " 380,\n",
       " 830,\n",
       " 170,\n",
       " 1040,\n",
       " 425,\n",
       " 1400,\n",
       " 165,\n",
       " 165,\n",
       " 165,\n",
       " 145,\n",
       " 590,\n",
       " 95,\n",
       " 1315,\n",
       " 615,\n",
       " 175,\n",
       " 735,\n",
       " 425,\n",
       " 155,\n",
       " 85,\n",
       " 845,\n",
       " 195,\n",
       " 330,\n",
       " 1410,\n",
       " 325,\n",
       " 95,\n",
       " 80,\n",
       " 170,\n",
       " 175,\n",
       " 1485,\n",
       " 700,\n",
       " 455,\n",
       " 1255,\n",
       " 185,\n",
       " 1410,\n",
       " 1295,\n",
       " 180,\n",
       " 615,\n",
       " 180,\n",
       " 560,\n",
       " 635,\n",
       " 1255,\n",
       " 370,\n",
       " 800,\n",
       " 885,\n",
       " 800,\n",
       " 1325,\n",
       " 570,\n",
       " 365,\n",
       " 160,\n",
       " 780,\n",
       " 740,\n",
       " 740,\n",
       " 965,\n",
       " 535,\n",
       " 390,\n",
       " 90,\n",
       " 170,\n",
       " 170,\n",
       " 195,\n",
       " 455,\n",
       " 85,\n",
       " 480,\n",
       " 1595,\n",
       " 870,\n",
       " 395,\n",
       " 845,\n",
       " 520,\n",
       " 310,\n",
       " 1350,\n",
       " 675,\n",
       " 450,\n",
       " 170,\n",
       " 135,\n",
       " 935,\n",
       " 350,\n",
       " 330,\n",
       " 315,\n",
       " 1220,\n",
       " 1435,\n",
       " 275,\n",
       " 335,\n",
       " 730,\n",
       " 85,\n",
       " 240,\n",
       " 170,\n",
       " 90,\n",
       " 1500,\n",
       " 870,\n",
       " 435,\n",
       " 965,\n",
       " 360,\n",
       " 315,\n",
       " 615,\n",
       " 720,\n",
       " 1550,\n",
       " 90,\n",
       " 1330,\n",
       " 95,\n",
       " 145,\n",
       " 315,\n",
       " 615,\n",
       " 455,\n",
       " 1215,\n",
       " 1100,\n",
       " 155,\n",
       " 615,\n",
       " 625,\n",
       " 185,\n",
       " 845,\n",
       " 835,\n",
       " 245,\n",
       " 245,\n",
       " 745,\n",
       " 565,\n",
       " 135,\n",
       " 355,\n",
       " 480,\n",
       " 165,\n",
       " 865,\n",
       " 1530,\n",
       " 395,\n",
       " 220,\n",
       " 155,\n",
       " 590,\n",
       " 165,\n",
       " 675,\n",
       " 435,\n",
       " 435,\n",
       " 475,\n",
       " 165,\n",
       " 1005,\n",
       " 315,\n",
       " 1995,\n",
       " 175,\n",
       " 1535,\n",
       " 140,\n",
       " 330,\n",
       " 165,\n",
       " 770,\n",
       " 170,\n",
       " 185,\n",
       " 1585,\n",
       " 800,\n",
       " 1075,\n",
       " 180,\n",
       " 1450,\n",
       " 160,\n",
       " 1100,\n",
       " 375,\n",
       " 735,\n",
       " 220,\n",
       " 1775,\n",
       " 175,\n",
       " 845,\n",
       " 245,\n",
       " 320,\n",
       " 760,\n",
       " 450,\n",
       " 155,\n",
       " 1165,\n",
       " 155,\n",
       " 145,\n",
       " 395,\n",
       " 525,\n",
       " 1255,\n",
       " 1625,\n",
       " 1245,\n",
       " 165,\n",
       " 535,\n",
       " 315,\n",
       " 705,\n",
       " 945,\n",
       " 935,\n",
       " 925,\n",
       " 1430,\n",
       " 965,\n",
       " 190,\n",
       " 815,\n",
       " 540,\n",
       " 175,\n",
       " 1165,\n",
       " 90,\n",
       " 825,\n",
       " 150,\n",
       " 1730,\n",
       " 530,\n",
       " 345,\n",
       " 2135,\n",
       " 380,\n",
       " 400,\n",
       " 585,\n",
       " 385,\n",
       " 1200,\n",
       " 1585,\n",
       " 170,\n",
       " 600,\n",
       " 315,\n",
       " 330,\n",
       " 170,\n",
       " 385,\n",
       " 140,\n",
       " 145,\n",
       " 1820,\n",
       " 1210,\n",
       " 385,\n",
       " 385,\n",
       " 150,\n",
       " 140,\n",
       " 510,\n",
       " 810,\n",
       " 510,\n",
       " 865,\n",
       " 975,\n",
       " 380,\n",
       " 170,\n",
       " 1065,\n",
       " 955,\n",
       " 150,\n",
       " 590,\n",
       " 625,\n",
       " 180,\n",
       " 830,\n",
       " 185,\n",
       " 810,\n",
       " 330,\n",
       " 985,\n",
       " 295,\n",
       " 165,\n",
       " 170,\n",
       " 440,\n",
       " 180,\n",
       " 165,\n",
       " 140,\n",
       " 1180,\n",
       " 985,\n",
       " 165,\n",
       " 1375,\n",
       " 80,\n",
       " 705,\n",
       " 535,\n",
       " 800,\n",
       " 170,\n",
       " 575,\n",
       " 165,\n",
       " 755,\n",
       " 175,\n",
       " 1315,\n",
       " 160,\n",
       " 465,\n",
       " 615,\n",
       " 1265,\n",
       " 370,\n",
       " 1600,\n",
       " 505,\n",
       " 2240,\n",
       " 220,\n",
       " 440,\n",
       " 505,\n",
       " 1270,\n",
       " 165,\n",
       " 170,\n",
       " 1760,\n",
       " 1375,\n",
       " 140,\n",
       " 870,\n",
       " 90,\n",
       " 295,\n",
       " 175,\n",
       " 550,\n",
       " 330,\n",
       " 1625,\n",
       " 800,\n",
       " 505,\n",
       " 1350,\n",
       " 755,\n",
       " 140,\n",
       " 385,\n",
       " 660,\n",
       " 540,\n",
       " 510,\n",
       " 785,\n",
       " 165,\n",
       " 705,\n",
       " 780,\n",
       " 90,\n",
       " 730,\n",
       " 300,\n",
       " 1780,\n",
       " 640,\n",
       " 625,\n",
       " 1200,\n",
       " 605,\n",
       " 435,\n",
       " 825,\n",
       " 145,\n",
       " 170,\n",
       " 165,\n",
       " 805,\n",
       " 1655,\n",
       " 745,\n",
       " 1295,\n",
       " 1700,\n",
       " 1075,\n",
       " 530,\n",
       " 90,\n",
       " 1560,\n",
       " 620,\n",
       " 875,\n",
       " 180,\n",
       " 140,\n",
       " 175,\n",
       " 165,\n",
       " 1545,\n",
       " 690,\n",
       " 155,\n",
       " 180,\n",
       " 700,\n",
       " 175,\n",
       " 270,\n",
       " 690,\n",
       " 560,\n",
       " 170,\n",
       " 435,\n",
       " 170,\n",
       " 1580,\n",
       " 290,\n",
       " 155,\n",
       " 170,\n",
       " 505,\n",
       " 590,\n",
       " 1100,\n",
       " 685,\n",
       " 745,\n",
       " 525,\n",
       " 140,\n",
       " 715,\n",
       " 1580,\n",
       " 160,\n",
       " 500,\n",
       " 335,\n",
       " 665,\n",
       " 1220,\n",
       " 145,\n",
       " 840,\n",
       " 1655,\n",
       " 1995,\n",
       " 150,\n",
       " 810,\n",
       " 390,\n",
       " 165,\n",
       " 165,\n",
       " 175,\n",
       " 1760,\n",
       " 1280,\n",
       " 165,\n",
       " 175,\n",
       " 1665,\n",
       " 790,\n",
       " 170,\n",
       " 145,\n",
       " 485,\n",
       " 160,\n",
       " 175,\n",
       " 1350,\n",
       " 1485,\n",
       " 455,\n",
       " 165,\n",
       " 1275,\n",
       " 170,\n",
       " 960,\n",
       " 750,\n",
       " 1195,\n",
       " 800,\n",
       " 1665,\n",
       " 145,\n",
       " 155,\n",
       " 75,\n",
       " 175,\n",
       " 1410,\n",
       " 705,\n",
       " 185,\n",
       " 155,\n",
       " 465,\n",
       " 660,\n",
       " 585,\n",
       " 780,\n",
       " 135,\n",
       " 355,\n",
       " 165,\n",
       " 665,\n",
       " 150,\n",
       " 165,\n",
       " 320,\n",
       " 1490,\n",
       " 585,\n",
       " 925,\n",
       " 270,\n",
       " 1375,\n",
       " 195,\n",
       " 340,\n",
       " 135,\n",
       " 170,\n",
       " 435,\n",
       " 465,\n",
       " 165,\n",
       " 1055,\n",
       " 300,\n",
       " 910,\n",
       " 505,\n",
       " 170,\n",
       " 140,\n",
       " 540,\n",
       " 325,\n",
       " 340,\n",
       " 170,\n",
       " 600,\n",
       " 560,\n",
       " 180,\n",
       " 2025,\n",
       " 175,\n",
       " 630,\n",
       " 400,\n",
       " 730,\n",
       " 160,\n",
       " 410,\n",
       " 165,\n",
       " 300,\n",
       " 1005,\n",
       " 800,\n",
       " 310,\n",
       " 85,\n",
       " 505,\n",
       " 155,\n",
       " 1365,\n",
       " 610,\n",
       " 165,\n",
       " 2135,\n",
       " 685,\n",
       " 165,\n",
       " 585,\n",
       " 150,\n",
       " 80,\n",
       " 1345,\n",
       " 705,\n",
       " 90,\n",
       " 1125,\n",
       " 90,\n",
       " 800,\n",
       " 735,\n",
       " 155,\n",
       " 570,\n",
       " 1450,\n",
       " 290,\n",
       " 915,\n",
       " 170,\n",
       " 1440,\n",
       " 1570,\n",
       " 170,\n",
       " 180,\n",
       " 955,\n",
       " 710,\n",
       " 145,\n",
       " 560,\n",
       " 175,\n",
       " 90,\n",
       " 510,\n",
       " 165,\n",
       " 1390,\n",
       " ...]"
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import re\n",
    "new_dur = []\n",
    "duration = test_data[\"Duration\"]\n",
    "overnight = test_data[\"Overnight_Flight\"]\n",
    "\n",
    "for string in duration:\n",
    "    s = str(string)\n",
    "    a = []\n",
    "    a = re.findall(r'\\d+', s)\n",
    "    if len(a)==2:\n",
    "        new_dur.append((int(a[0])*60)+int(a[1]))\n",
    "    else:\n",
    "        new_dur.append(int(a[0])*60)\n",
    "\n",
    "new_dur"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "stops = test_data[\"Total_Stops\"]\n",
    "new_Stops = []\n",
    "for i in stops:\n",
    "    if i==\"non-stop\":\n",
    "        new_Stops.append(0)\n",
    "    elif i==\"1 stop\":\n",
    "        new_Stops.append(1)\n",
    "    elif i==\"2 stops\":\n",
    "        new_Stops.append(2)\n",
    "    else:\n",
    "        new_Stops.append(3)\n",
    "\n",
    "test_data = test_data.drop([\"Dep_Time\",\"Arrival_Time\",\"Duration\",\"Route\",\"Total_Stops\"],axis=1)\n",
    "test_data[\"Duration\"] = new_dur\n",
    "test_data[\"Total_Stops\"] = new_Stops"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "from datetime import date\n",
    "no_of_days = []\n",
    "temp_date = test_data[\"Date_of_Journey\"]\n",
    "weekday = []\n",
    "\n",
    "for i in temp_date:\n",
    "    s = i.split(\"/\")\n",
    "    year = int(s[2])\n",
    "    month = int(s[1])\n",
    "    day = int(s[0])\n",
    "    \n",
    "    x = date(year,month,day)\n",
    "    y = date.today()\n",
    "    \n",
    "    weekday.append(x.weekday())\n",
    "\n",
    "    no_of_days.append((x-y).days)\n",
    "   \n",
    "test_data[\"No_of_days\"] = no_of_days\n",
    "test_data[\"Weekday\"] = weekday"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "day = test_data[\"Weekday\"]\n",
    "expensive_ticket_day = []\n",
    "\n",
    "for d in day:\n",
    "    if d==4 or d==5:\n",
    "        expensive_ticket_day.append(1)\n",
    "    else:\n",
    "        expensive_ticket_day.append(0)\n",
    "        \n",
    "test_data[\"Expensive_Day\"] = expensive_ticket_day"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_data = test_data.drop([\"Date_of_Journey\"],axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "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>Airline</th>\n",
       "      <th>Source</th>\n",
       "      <th>Destination</th>\n",
       "      <th>No_of_days</th>\n",
       "      <th>Weekday</th>\n",
       "      <th>Expensive_Day</th>\n",
       "      <th>Overnight_Flight</th>\n",
       "      <th>Duration</th>\n",
       "      <th>Total_Stops</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>91</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>655</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>66</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>240</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>75</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1425</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>6</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>75</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>780</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>109</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>170</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Airline  Source  Destination  No_of_days  Weekday  Expensive_Day  \\\n",
       "0        4       2            1          91        3              0   \n",
       "1        3       3            0          66        6              0   \n",
       "2        4       2            1          75        1              0   \n",
       "3        6       2            1          75        1              0   \n",
       "4        0       0            2         109        0              0   \n",
       "\n",
       "   Overnight_Flight  Duration  Total_Stops  \n",
       "0                 1       655            1  \n",
       "1                 0       240            1  \n",
       "2                 1      1425            1  \n",
       "3                 0       780            1  \n",
       "4                 1       170            0  "
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "of = test_data[\"Overnight_Flight\"]\n",
    "d = test_data[\"Duration\"]\n",
    "ts = test_data[\"Total_Stops\"]\n",
    "test_data = test_data.drop([\"Overnight_Flight\",\"Duration\",\"Total_Stops\"],axis=1)\n",
    "\n",
    "test_data[\"Overnight_Flight\"] = of\n",
    "test_data[\"Duration\"] = d\n",
    "test_data[\"Total_Stops\"] = ts\n",
    "\n",
    "test_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "# test_data.to_csv(\"testing_Data.csv\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_data = test_data.drop([\"Expensive_Day\"],axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([10262. ,  5276. , 14488.5, ..., 13843. , 16666. ,  7155. ])"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "price = regressor.predict(test_data)\n",
    "price"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[10262.   5276.  14488.5 ... 13843.  16666.   7155. ]\n"
     ]
    }
   ],
   "source": [
    "print(price)"
   ]
  },
  {
   "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>Price</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>10262.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>5276.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>14488.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>14277.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3383.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     Price\n",
       "0  10262.0\n",
       "1   5276.0\n",
       "2  14488.5\n",
       "3  14277.0\n",
       "4   3383.0"
      ]
     },
     "execution_count": 31,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = pd.DataFrame({\"Price\":price})\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "df.to_excel(\"attempt_6.xlsx\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.15"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}