0. Setting Our Data Up.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import json\n",
    "import matplotlib.pyplot as plt\n",
    "import matplotlib.lines as mlines\n",
    "import matplotlib.transforms as mtransforms\n",
    "import numpy as np\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.linear_model import LogisticRegression\n",
    "\n",
    "from sklearn.model_selection import train_test_split, KFold, cross_val_score, RandomizedSearchCV\n",
    "from sklearn.metrics import log_loss\n",
    "from sklearn.calibration import calibration_curve, CalibratedClassifierCV\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "datasets=['England', 'European_Championship', 'France', 'Germany', 'Spain', 'Italy', 'World_Cup']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "event_dataframes = []\n",
    "match_dataframes =  []\n",
    "for dataset in datasets:\n",
    "    with open('Wyscout/events_'+dataset+'.json') as f:\n",
    "        json_data = json.load(f)\n",
    "        pandas_data = pd.DataFrame(json_data)\n",
    "        event_dataframes.append(pandas_data)\n",
    "    with open('Wyscout/matches_'+dataset+'.json') as f:\n",
    "        json_data = json.load(f)\n",
    "        pandas_data = pd.DataFrame(json_data)\n",
    "        match_dataframes.append(pandas_data)\n",
    "\n",
    "\n",
    "all_events_df = pd.concat(event_dataframes, axis=0).reset_index(drop=True)\n",
    "matches_df = pd.concat(match_dataframes, axis=0).reset_index(drop=True)\n",
    "with open('Wyscout/players.json') as f:\n",
    "    player_json=json.load(f)\n",
    "    player_df = pd.DataFrame(player_json)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "all_events_df=pd.merge(all_events_df, player_df[['wyId', 'foot', 'firstName', 'lastName']], \n",
    "                       left_on='playerId', right_on='wyId')\n",
    "all_events_df=pd.merge(all_events_df, matches_df[['wyId', 'label', 'venue', 'date']], left_on='matchId', right_on='wyId')\n",
    "all_events_df=all_events_df.drop(columns=['wyId_x', 'wyId_y'])\n",
    "all_events_df=all_events_df.sort_values(['matchId', 'matchPeriod', 'eventSec'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "all_events_df['previous_event'] = all_events_df['subEventName'].shift(1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "all_shots = all_events_df[(all_events_df['subEventName'] == 'Shot') |(all_events_df['subEventName'] =='Free kick shot')].copy()\n",
    "\n",
    "all_shots['free_kick'] = 1*(all_shots['subEventName'] == 'Free kick shot')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "all_shots['rebound']= 1*(all_shots['previous_event'] == 'Penalty') \\\n",
    "                    + 1*(all_shots['previous_event'] == 'Free kick shot') \\\n",
    "                    + 1*(all_shots['previous_event'] == 'Shot') \\\n",
    "                    + 1*(all_shots['previous_event'] == 'Save attempt') \n",
    "\n",
    "all_shots['prev_cross'] = 1*(all_shots['previous_event'] == 'Corner') \\\n",
    "                        + 1*(all_shots['previous_event'] == 'Free kick cross') \\\n",
    "                        + 1*(all_shots['previous_event'] == 'Cross') \n",
    "\n",
    "all_shots['prev_touch'] = 1*(all_shots['previous_event'] == 'Touch')\n",
    "\n",
    "all_shots['prev_pass'] = 1*(all_shots['previous_event'] == 'Simple pass') \\\n",
    "                       + 1*(all_shots['previous_event'] == 'Head pass') \\\n",
    "                       + 1*(all_shots['previous_event'] == 'Goal kick') \\\n",
    "\n",
    "all_shots['prev_smart_pass'] = 1*(all_shots['previous_event'] == 'Smart pass')\n",
    "\n",
    "all_shots['prev_duel'] = 1*(all_shots['previous_event'] == 'Air duel') \\\n",
    "                       + 1*(all_shots['previous_event'] == 'Ground defending duel')  \\\n",
    "                       + 1*(all_shots['previous_event'] == 'Ground attacking duel') \\\n",
    "                       + 1*(all_shots['previous_event'] == 'Ground loose ball duel duel') \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "shots_model=pd.DataFrame(columns=['Goal','X','Y', 'side_of_field', 'left_foot', \n",
    "                                  'right_foot', 'header', 'counter_attack', 'strong_foot'])"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Note, this code takes a while to run. If you have any suggestions to improve this, please submit a PR."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Special thanks to David Sumpter for the code to determine the distance and angle from the goal"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "for i,shot in all_shots.iterrows():\n",
    "    shots_model.at[i,'X']=100-shot['positions'][0]['x']\n",
    "    shots_model.at[i,'Y']=shot['positions'][0]['y']\n",
    "    shots_model.at[i,'side_of_field']= 1*(shot['positions'][0]['y'] <  50)\n",
    "    shots_model.at[i,'C']=abs(shot['positions'][0]['y']-50)\n",
    "    \n",
    "    #Distance in metres and shot angle in radians.\n",
    "    x=shots_model.at[i,'X']*105/100\n",
    "    y=shots_model.at[i,'C']*65/100\n",
    "    shots_model.at[i,'Distance']=np.sqrt(x**2 + y**2)\n",
    "    a = np.arctan(7.32 *x /(x**2 + y**2 - (7.32/2)**2))\n",
    "    if a<0:\n",
    "        a=np.pi+a\n",
    "    shots_model.at[i,'Angle'] =a\n",
    "    shottags=[tag['id'] for tag in shot['tags']]\n",
    "    if 101 in shottags:\n",
    "        shots_model.at[i,'Goal']=1\n",
    "    if 401 in shottags:\n",
    "        shots_model.at[i, 'left_foot']=1\n",
    "        if shot.loc['foot']=='left':\n",
    "            shots_model.at[i, 'strong_foot'] = 1\n",
    "    if 402 in shottags:\n",
    "        shots_model.at[i, 'right_foot']=1\n",
    "        if shot.loc['foot']=='right':\n",
    "            shots_model.at[i, 'strong_foot'] = 1\n",
    "    if 403 in shottags:\n",
    "        shots_model.at[i, 'header']=1\n",
    "    if 1901 in shottags:\n",
    "        shots_model.at[i, 'counter_attack'] = 1\n",
    "shots_model = shots_model.fillna(0)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "shots_model['out_swinging'] = 1*(shots_model['side_of_field'] == 0)*(shots_model['right_foot'] == 1) \\\n",
    "                           + 1*(shots_model['side_of_field'] == 1)*(shots_model['left_foot'] == 1)\n",
    "shots_model['in_swinging'] = 1*(shots_model['side_of_field'] == 0)*(shots_model['left_foot'] == 1) \\\n",
    "                            + 1*(shots_model['side_of_field'] == 1)*(shots_model['right_foot'] == 1)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "shots_model = pd.merge(shots_model, all_shots[['rebound', 'prev_cross', 'prev_touch',  'prev_pass',\n",
    "                                               'prev_smart_pass',  'free_kick', 'prev_duel',  'firstName', \n",
    "                                               'lastName', 'label', 'venue', 'date','eventSec', 'matchPeriod']], \n",
    "                       left_index=True, right_index=True, how='left')\n",
    "shots_model = shots_model.reset_index(drop=True)\n"
   ]
  },
  {
   "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>Goal</th>\n",
       "      <th>X</th>\n",
       "      <th>Y</th>\n",
       "      <th>side_of_field</th>\n",
       "      <th>left_foot</th>\n",
       "      <th>right_foot</th>\n",
       "      <th>header</th>\n",
       "      <th>counter_attack</th>\n",
       "      <th>strong_foot</th>\n",
       "      <th>C</th>\n",
       "      <th>...</th>\n",
       "      <th>prev_smart_pass</th>\n",
       "      <th>free_kick</th>\n",
       "      <th>prev_duel</th>\n",
       "      <th>firstName</th>\n",
       "      <th>lastName</th>\n",
       "      <th>label</th>\n",
       "      <th>venue</th>\n",
       "      <th>date</th>\n",
       "      <th>eventSec</th>\n",
       "      <th>matchPeriod</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>9</td>\n",
       "      <td>29</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Blaise</td>\n",
       "      <td>Matuidi</td>\n",
       "      <td>France - Romania, 2 - 1</td>\n",
       "      <td>Stade de France</td>\n",
       "      <td>June 10, 2016 at 9:00:00 PM GMT+2</td>\n",
       "      <td>31.226217</td>\n",
       "      <td>1H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>29</td>\n",
       "      <td>29</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>21.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Mihai Doru</td>\n",
       "      <td>Pintilii</td>\n",
       "      <td>France - Romania, 2 - 1</td>\n",
       "      <td>Stade de France</td>\n",
       "      <td>June 10, 2016 at 9:00:00 PM GMT+2</td>\n",
       "      <td>143.119551</td>\n",
       "      <td>1H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>57</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>Bogdan Sorin</td>\n",
       "      <td>Stancu</td>\n",
       "      <td>France - Romania, 2 - 1</td>\n",
       "      <td>Stade de France</td>\n",
       "      <td>June 10, 2016 at 9:00:00 PM GMT+2</td>\n",
       "      <td>219.576026</td>\n",
       "      <td>1H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>61</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>Florin</td>\n",
       "      <td>Andone</td>\n",
       "      <td>France - Romania, 2 - 1</td>\n",
       "      <td>Stade de France</td>\n",
       "      <td>June 10, 2016 at 9:00:00 PM GMT+2</td>\n",
       "      <td>247.532561</td>\n",
       "      <td>1H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>25</td>\n",
       "      <td>33</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>17.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Antoine</td>\n",
       "      <td>Griezmann</td>\n",
       "      <td>France - Romania, 2 - 1</td>\n",
       "      <td>Stade de France</td>\n",
       "      <td>June 10, 2016 at 9:00:00 PM GMT+2</td>\n",
       "      <td>557.319065</td>\n",
       "      <td>1H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45279</th>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>45</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Diego Sebasti\\u00e1n</td>\n",
       "      <td>Laxalt Su\\u00e1rez</td>\n",
       "      <td>Genoa - Torino, 1 - 2</td>\n",
       "      <td></td>\n",
       "      <td>May 20, 2018 at 3:00:00 PM GMT+2</td>\n",
       "      <td>1152.032980</td>\n",
       "      <td>2H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45280</th>\n",
       "      <td>0</td>\n",
       "      <td>7</td>\n",
       "      <td>38</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>12.0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Giuseppe</td>\n",
       "      <td>Rossi</td>\n",
       "      <td>Genoa - Torino, 1 - 2</td>\n",
       "      <td></td>\n",
       "      <td>May 20, 2018 at 3:00:00 PM GMT+2</td>\n",
       "      <td>1251.730517</td>\n",
       "      <td>2H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45281</th>\n",
       "      <td>1</td>\n",
       "      <td>10</td>\n",
       "      <td>46</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>4.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>Goran</td>\n",
       "      <td>Pandev</td>\n",
       "      <td>Genoa - Torino, 1 - 2</td>\n",
       "      <td></td>\n",
       "      <td>May 20, 2018 at 3:00:00 PM GMT+2</td>\n",
       "      <td>2065.034482</td>\n",
       "      <td>2H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45282</th>\n",
       "      <td>0</td>\n",
       "      <td>21</td>\n",
       "      <td>32</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>18.0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Stephane</td>\n",
       "      <td>Omeonga</td>\n",
       "      <td>Genoa - Torino, 1 - 2</td>\n",
       "      <td></td>\n",
       "      <td>May 20, 2018 at 3:00:00 PM GMT+2</td>\n",
       "      <td>2367.252041</td>\n",
       "      <td>2H</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45283</th>\n",
       "      <td>0</td>\n",
       "      <td>7</td>\n",
       "      <td>43</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Goran</td>\n",
       "      <td>Pandev</td>\n",
       "      <td>Genoa - Torino, 1 - 2</td>\n",
       "      <td></td>\n",
       "      <td>May 20, 2018 at 3:00:00 PM GMT+2</td>\n",
       "      <td>2579.867806</td>\n",
       "      <td>2H</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>45284 rows × 28 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       Goal   X   Y  side_of_field  left_foot  right_foot  header  \\\n",
       "0         0   9  29              1          0           1       0   \n",
       "1         0  29  29              1          0           1       0   \n",
       "2         0   4  57              0          0           1       0   \n",
       "3         0   4  61              0          0           0       1   \n",
       "4         0  25  33              1          1           0       0   \n",
       "...     ...  ..  ..            ...        ...         ...     ...   \n",
       "45279     0   5  45              1          1           0       0   \n",
       "45280     0   7  38              1          1           0       0   \n",
       "45281     1  10  46              1          1           0       0   \n",
       "45282     0  21  32              1          0           1       0   \n",
       "45283     0   7  43              1          1           0       0   \n",
       "\n",
       "       counter_attack  strong_foot     C  ...  prev_smart_pass  free_kick  \\\n",
       "0                   0            0  21.0  ...                0          0   \n",
       "1                   0            1  21.0  ...                0          0   \n",
       "2                   0            1   7.0  ...                0          0   \n",
       "3                   0            0  11.0  ...                0          0   \n",
       "4                   0            1  17.0  ...                0          0   \n",
       "...               ...          ...   ...  ...              ...        ...   \n",
       "45279               0            1   5.0  ...                0          0   \n",
       "45280               0            1  12.0  ...                1          0   \n",
       "45281               0            1   4.0  ...                0          0   \n",
       "45282               0            1  18.0  ...                0          0   \n",
       "45283               0            1   7.0  ...                1          0   \n",
       "\n",
       "       prev_duel             firstName            lastName  \\\n",
       "0              0                Blaise             Matuidi   \n",
       "1              0            Mihai Doru            Pintilii   \n",
       "2              1          Bogdan Sorin              Stancu   \n",
       "3              1                Florin              Andone   \n",
       "4              0               Antoine           Griezmann   \n",
       "...          ...                   ...                 ...   \n",
       "45279          0  Diego Sebasti\\u00e1n  Laxalt Su\\u00e1rez   \n",
       "45280          0              Giuseppe               Rossi   \n",
       "45281          1                 Goran              Pandev   \n",
       "45282          0              Stephane             Omeonga   \n",
       "45283          0                 Goran              Pandev   \n",
       "\n",
       "                         label            venue  \\\n",
       "0      France - Romania, 2 - 1  Stade de France   \n",
       "1      France - Romania, 2 - 1  Stade de France   \n",
       "2      France - Romania, 2 - 1  Stade de France   \n",
       "3      France - Romania, 2 - 1  Stade de France   \n",
       "4      France - Romania, 2 - 1  Stade de France   \n",
       "...                        ...              ...   \n",
       "45279    Genoa - Torino, 1 - 2                    \n",
       "45280    Genoa - Torino, 1 - 2                    \n",
       "45281    Genoa - Torino, 1 - 2                    \n",
       "45282    Genoa - Torino, 1 - 2                    \n",
       "45283    Genoa - Torino, 1 - 2                    \n",
       "\n",
       "                                    date     eventSec  matchPeriod  \n",
       "0      June 10, 2016 at 9:00:00 PM GMT+2    31.226217           1H  \n",
       "1      June 10, 2016 at 9:00:00 PM GMT+2   143.119551           1H  \n",
       "2      June 10, 2016 at 9:00:00 PM GMT+2   219.576026           1H  \n",
       "3      June 10, 2016 at 9:00:00 PM GMT+2   247.532561           1H  \n",
       "4      June 10, 2016 at 9:00:00 PM GMT+2   557.319065           1H  \n",
       "...                                  ...          ...          ...  \n",
       "45279   May 20, 2018 at 3:00:00 PM GMT+2  1152.032980           2H  \n",
       "45280   May 20, 2018 at 3:00:00 PM GMT+2  1251.730517           2H  \n",
       "45281   May 20, 2018 at 3:00:00 PM GMT+2  2065.034482           2H  \n",
       "45282   May 20, 2018 at 3:00:00 PM GMT+2  2367.252041           2H  \n",
       "45283   May 20, 2018 at 3:00:00 PM GMT+2  2579.867806           2H  \n",
       "\n",
       "[45284 rows x 28 columns]"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "shots_model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "shots_model.to_csv('shots_design_matrix.csv')"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}