Thomas/Darron/Mark Celebrating Cities Hack The Last Mile work

BaltimoreTravelOptions.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 59,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "# iPython stuff\n",
    "from __future__ import print_function\n",
    "from ipywidgets import interact, interactive, fixed\n",
    "from IPython.display import display\n",
    "from IPython.display import IFrame\n",
    "import ipywidgets as widgets\n",
    "\n",
    "# Geopy stuff\n",
    "from geopy.geocoders import Nominatim\n",
    "from geopy.distance import vincenty\n",
    "geolocator = Nominatim()\n",
    "\n",
    "# Darren Initialization Stuff\n",
    "from googlemaps import distance_matrix as dm\n",
    "import googlemaps\n",
    "import cPickle\n",
    "import pandas as pd\n",
    "\n",
    "google_api_key = 'SECRET'\n",
    "gmaps = googlemaps.Client(google_api_key)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 60,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "# Darren Code Stuff\n",
    "\n",
    "# def travel_options(start_lat,)\n",
    "\n",
    "lr_dests = pd.read_csv('data/Maryland_Transit__Light_Rail_Stops.csv')\n",
    "pd_dest_list = lr_dests.values.tolist()\n",
    "\n",
    "dest_list = []\n",
    "for stop in pd_dest_list:\n",
    "    dest_list.append((float(stop[0]), float(stop[1])))\n",
    "\n",
    "import uber_rides\n",
    "from uber_rides.client import UberRidesClient\n",
    "uber_server_token = 'SECRET'\n",
    "from uber_rides.session import Session\n",
    "session = Session(server_token=uber_server_token)\n",
    "client = UberRidesClient(session)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 61,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "def get_travel_options(start_lat, start_lon):\n",
    "\n",
    "    # set origin list to just the start location\n",
    "    pd_origin_list = [[start_lat,start_lon]]\n",
    "    origin_list = []\n",
    "    for orig in pd_origin_list:\n",
    "        print(orig)\n",
    "        origin_list.append((float(orig[0]), float(orig[1])))\n",
    "\n",
    "    # calculate matrix\n",
    "    walk_matrix = dm.distance_matrix(gmaps, origin_list[0:1], dest_list, mode='walking')\n",
    "    # rail_matrix = dm.distance_matrix(gmaps, origin_list[0:1], dest_list, mode='transit', transit_mode='rail')\n",
    "    bus_matrix = dm.distance_matrix(gmaps, origin_list[0:1], dest_list, mode='transit', transit_mode='bus')\n",
    "\n",
    "    travel_options = []\n",
    "    \n",
    "    # UberX\n",
    "    for s in origin_list:\n",
    "        for e in dest_list[0:1]:\n",
    "            ride = client.get_price_estimates(start_latitude=s[0], start_longitude=s[1], end_latitude=e[0], end_longitude=e[1])\n",
    "            for i, rd in enumerate(ride.json['prices']):\n",
    "                if rd['display_name'] == 'uberX':\n",
    "                    travel_options.append((s[0],s[1], e[0],e[1], 'uberX', rd['distance'], rd['high_estimate'], rd['duration']/60))\n",
    "\n",
    "    # Walk, Train, Bus\n",
    "    travel_modes = ['Walk', 'Bus']\n",
    "    travel_mx = [walk_matrix, bus_matrix]\n",
    "    for mx, mode in enumerate(travel_modes):\n",
    "        for i, s in enumerate(origin_list):\n",
    "            for j, e in enumerate(dest_list[0:1]):\n",
    "                travel_options.append(([s[0],s[1], e[0],e[1], mode, travel_mx[mx]['rows'][i]['elements'][j]['distance']['value']/1000*0.621371, 0, travel_mx[mx]['rows'][i]['elements'][j]['duration']['value']/60]))\n",
    "\n",
    "    return travel_options"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 62,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Lat/lon of the address #1 is: (39.376362, -76.664136)\n",
      "[39.376362, -76.664136]\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>distance</th>\n",
       "      <th>end</th>\n",
       "      <th>high_cost_est</th>\n",
       "      <th>mode</th>\n",
       "      <th>start</th>\n",
       "      <th>time_in_mins</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2.100000</td>\n",
       "      <td>(39.376072, -76.650423)</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>uberX</td>\n",
       "      <td>(39.376362, -76.664136)</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1.242742</td>\n",
       "      <td>(39.376072, -76.650423)</td>\n",
       "      <td>5.666667</td>\n",
       "      <td>Walk</td>\n",
       "      <td>(39.376362, -76.664136)</td>\n",
       "      <td>34</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1.864113</td>\n",
       "      <td>(39.376072, -76.650423)</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>Bus</td>\n",
       "      <td>(39.376362, -76.664136)</td>\n",
       "      <td>18</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   distance                      end  high_cost_est   mode  \\\n",
       "0  2.100000  (39.376072, -76.650423)       6.000000  uberX   \n",
       "1  1.242742  (39.376072, -76.650423)       5.666667   Walk   \n",
       "2  1.864113  (39.376072, -76.650423)       3.000000    Bus   \n",
       "\n",
       "                     start  time_in_mins  \n",
       "0  (39.376362, -76.664136)             6  \n",
       "1  (39.376362, -76.664136)            34  \n",
       "2  (39.376362, -76.664136)            18  "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "# This function gets called everytime an input changes\n",
    "def input_function(address1, hourly_rate):\n",
    "    # Location 1\n",
    "    location1 = geolocator.geocode(address1)\n",
    "    print(\"Lat/lon of the address #1 is: ({}, {})\".format(location1.latitude,location1.longitude))\n",
    "    travel_options = get_travel_options(location1.latitude, location1.longitude)\n",
    "    travel_options_data = []\n",
    "    for option in travel_options:\n",
    "        try: \n",
    "            data = {\n",
    "                'start': (option[0],option[1]),\n",
    "                'end': (option[2],option[3]),\n",
    "                'mode': option[4],\n",
    "                'distance': option[5],\n",
    "                'high_cost_est': option[6],\n",
    "                'time_in_mins': option[7]\n",
    "            }\n",
    "            if(data['mode'] != \"uberX\"):\n",
    "                data['high_cost_est'] = (float(data['time_in_mins'])/60)*int(hourly_rate)\n",
    "            travel_options_data.append(data)\n",
    "        except: \n",
    "            print()\n",
    "            # nothing, it's a hack.\n",
    "            \n",
    "    df = pd.DataFrame(travel_options_data)\n",
    "    display(df)\n",
    "\n",
    "    # return {'address': address1, 'hourly_rate': hourly_rate}\n",
    "\n",
    "w = interactive(input_function, address1='2006 Smith Ave, Baltimore, MD 21209', hourly_rate=widgets.IntSlider(min=0,max=100,step=5,value=10))\n",
    "\n",
    "display(w)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 63,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "        <iframe\n",
       "            width=\"700\"\n",
       "            height=\"700\"\n",
       "            src=\"https://www.google.com/maps/embed/v1/directions?key=SECRET&origin=Oslo+Norway&destination=Telemark+Norway&avoid=tolls|highways\"\n",
       "            frameborder=\"0\"\n",
       "            allowfullscreen\n",
       "        ></iframe>\n",
       "        "
      ],
      "text/plain": [
       "<IPython.lib.display.IFrame at 0x10877bfd0>"
      ]
     },
     "execution_count": 63,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Display Map\n",
    "from IPython.display import IFrame\n",
    "IFrame('https://www.google.com/maps/embed/v1/directions?key=SECRET&origin=Oslo+Norway&destination=Telemark+Norway&avoid=tolls|highways', width=700, height=700)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "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.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}

z_Maryland_Transit__Light_Rail_Stops.csv