icehockeygames_dataprep.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Prepare icehockey data from swehockey"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "\n",
    "import datetime\n",
    "\n",
    "import swehockey.swehockey_scraper as swe "
   ]
  },
  {
   "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>schedule_id</th>\n",
       "      <th>league</th>\n",
       "      <th>season</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>11450</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10371</td>\n",
       "      <td>shl</td>\n",
       "      <td>2019/20</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>9171</td>\n",
       "      <td>shl</td>\n",
       "      <td>2018/19</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>8121</td>\n",
       "      <td>shl</td>\n",
       "      <td>2017/18</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>7132</td>\n",
       "      <td>shl</td>\n",
       "      <td>2016/17</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  schedule_id league   season\n",
       "0       11450    shl  2020/21\n",
       "1       10371    shl  2019/20\n",
       "2        9171    shl  2018/19\n",
       "3        8121    shl  2017/18\n",
       "4        7132    shl  2016/17"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Read in needed schedule-ids\n",
    "df_scheduleid = pd.read_csv(\"https://raw.githubusercontent.com/msjoelin/swehockey_scraper/master/data/scheduleid.csv\", \n",
    "                           error_bad_lines=False,\n",
    "                          dtype=str)\n",
    "\n",
    "df_scheduleid.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>date</th>\n",
       "      <th>game</th>\n",
       "      <th>score</th>\n",
       "      <th>periodscore</th>\n",
       "      <th>spectators</th>\n",
       "      <th>schedule_id</th>\n",
       "      <th>game_id</th>\n",
       "      <th>league</th>\n",
       "      <th>season</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Leksands IF  - Skellefteå AIK</td>\n",
       "      <td>4 - 5</td>\n",
       "      <td>(1-2, 0-2, 3-0, 0-1)</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490388</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Frölunda HC  - HV 71</td>\n",
       "      <td>3 - 0</td>\n",
       "      <td>(2-0, 1-0, 0-0)</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490391</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Luleå HF  - Färjestad BK</td>\n",
       "      <td>4 - 2</td>\n",
       "      <td>(2-0, 1-0, 1-2)</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490392</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Rögle BK  - Linköping HC</td>\n",
       "      <td>6 - 4</td>\n",
       "      <td>(1-1, 3-1, 2-2)</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490394</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2020-09-22</td>\n",
       "      <td>IF Malmö Redhawks  - Leksands IF</td>\n",
       "      <td>3 - 5</td>\n",
       "      <td>(1-3, 1-1, 1-1)</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490412</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "         date                              game  score           periodscore  \\\n",
       "0  2020-09-19     Leksands IF  - Skellefteå AIK  4 - 5  (1-2, 0-2, 3-0, 0-1)   \n",
       "1  2020-09-19              Frölunda HC  - HV 71  3 - 0       (2-0, 1-0, 0-0)   \n",
       "2  2020-09-19          Luleå HF  - Färjestad BK  4 - 2       (2-0, 1-0, 1-2)   \n",
       "3  2020-09-19          Rögle BK  - Linköping HC  6 - 4       (1-1, 3-1, 2-2)   \n",
       "4  2020-09-22  IF Malmö Redhawks  - Leksands IF  3 - 5       (1-3, 1-1, 1-1)   \n",
       "\n",
       "   spectators schedule_id game_id league   season  \n",
       "0        50.0       11450  490388    shl  2020/21  \n",
       "1        50.0       11450  490391    shl  2020/21  \n",
       "2        50.0       11450  490392    shl  2020/21  \n",
       "3        50.0       11450  490394    shl  2020/21  \n",
       "4        50.0       11450  490412    shl  2020/21  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# get games for schedule ids\n",
    "games = swe.getGames(df_scheduleid)\n",
    "games.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>date</th>\n",
       "      <th>game</th>\n",
       "      <th>score</th>\n",
       "      <th>periodscore</th>\n",
       "      <th>spectators</th>\n",
       "      <th>schedule_id</th>\n",
       "      <th>game_id</th>\n",
       "      <th>league</th>\n",
       "      <th>season</th>\n",
       "      <th>home</th>\n",
       "      <th>...</th>\n",
       "      <th>p4score_home</th>\n",
       "      <th>p4score_away</th>\n",
       "      <th>p5score_home</th>\n",
       "      <th>p5score_away</th>\n",
       "      <th>result</th>\n",
       "      <th>result_p1</th>\n",
       "      <th>result_p2</th>\n",
       "      <th>result_p3</th>\n",
       "      <th>result_p4</th>\n",
       "      <th>result_p5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Leksands IF  - Skellefteå AIK</td>\n",
       "      <td>4 - 5</td>\n",
       "      <td>1-2, 0-2, 3-0, 0-1</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490388</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "      <td>Leksands IF</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>draw</td>\n",
       "      <td>away</td>\n",
       "      <td>away</td>\n",
       "      <td>home</td>\n",
       "      <td>away</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Frölunda HC  - HV 71</td>\n",
       "      <td>3 - 0</td>\n",
       "      <td>2-0, 1-0, 0-0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490391</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "      <td>Frölunda HC</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>home</td>\n",
       "      <td>home</td>\n",
       "      <td>home</td>\n",
       "      <td>draw</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Luleå HF  - Färjestad BK</td>\n",
       "      <td>4 - 2</td>\n",
       "      <td>2-0, 1-0, 1-2</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490392</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "      <td>Luleå HF</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>home</td>\n",
       "      <td>home</td>\n",
       "      <td>home</td>\n",
       "      <td>away</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2020-09-19</td>\n",
       "      <td>Rögle BK  - Linköping HC</td>\n",
       "      <td>6 - 4</td>\n",
       "      <td>1-1, 3-1, 2-2</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490394</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "      <td>Rögle BK</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>home</td>\n",
       "      <td>draw</td>\n",
       "      <td>home</td>\n",
       "      <td>draw</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2020-09-22</td>\n",
       "      <td>IF Malmö Redhawks  - Leksands IF</td>\n",
       "      <td>3 - 5</td>\n",
       "      <td>1-3, 1-1, 1-1</td>\n",
       "      <td>50.0</td>\n",
       "      <td>11450</td>\n",
       "      <td>490412</td>\n",
       "      <td>shl</td>\n",
       "      <td>2020/21</td>\n",
       "      <td>IF Malmö Redhawks</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>away</td>\n",
       "      <td>away</td>\n",
       "      <td>draw</td>\n",
       "      <td>draw</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 34 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "         date                              game  score         periodscore  \\\n",
       "0  2020-09-19     Leksands IF  - Skellefteå AIK  4 - 5  1-2, 0-2, 3-0, 0-1   \n",
       "1  2020-09-19              Frölunda HC  - HV 71  3 - 0       2-0, 1-0, 0-0   \n",
       "2  2020-09-19          Luleå HF  - Färjestad BK  4 - 2       2-0, 1-0, 1-2   \n",
       "3  2020-09-19          Rögle BK  - Linköping HC  6 - 4       1-1, 3-1, 2-2   \n",
       "4  2020-09-22  IF Malmö Redhawks  - Leksands IF  3 - 5       1-3, 1-1, 1-1   \n",
       "\n",
       "   spectators schedule_id game_id league   season               home  ...  \\\n",
       "0        50.0       11450  490388    shl  2020/21        Leksands IF  ...   \n",
       "1        50.0       11450  490391    shl  2020/21        Frölunda HC  ...   \n",
       "2        50.0       11450  490392    shl  2020/21           Luleå HF  ...   \n",
       "3        50.0       11450  490394    shl  2020/21           Rögle BK  ...   \n",
       "4        50.0       11450  490412    shl  2020/21  IF Malmö Redhawks  ...   \n",
       "\n",
       "  p4score_home  p4score_away  p5score_home p5score_away result result_p1  \\\n",
       "0          0.0           1.0           NaN          NaN   draw      away   \n",
       "1          NaN           NaN           NaN          NaN   home      home   \n",
       "2          NaN           NaN           NaN          NaN   home      home   \n",
       "3          NaN           NaN           NaN          NaN   home      draw   \n",
       "4          NaN           NaN           NaN          NaN   away      away   \n",
       "\n",
       "  result_p2 result_p3  result_p4  result_p5  \n",
       "0      away      home       away       None  \n",
       "1      home      draw       None       None  \n",
       "2      home      away       None       None  \n",
       "3      home      draw       None       None  \n",
       "4      draw      draw       None       None  \n",
       "\n",
       "[5 rows x 34 columns]"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Clean up the output with cleanGames\n",
    "df_games_clean = swe.cleanGames(games.copy())\n",
    "df_games_clean.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "df_games_clean.to_csv(\"data/df_games.csv\", index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "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>date</th>\n",
       "      <th>game</th>\n",
       "      <th>score</th>\n",
       "      <th>periodscore</th>\n",
       "      <th>spectators</th>\n",
       "      <th>schedule_id</th>\n",
       "      <th>game_id</th>\n",
       "      <th>league</th>\n",
       "      <th>season</th>\n",
       "      <th>team</th>\n",
       "      <th>...</th>\n",
       "      <th>H2H_W</th>\n",
       "      <th>H2H_D</th>\n",
       "      <th>H2H_L</th>\n",
       "      <th>points</th>\n",
       "      <th>points_cum</th>\n",
       "      <th>points_cum_prev</th>\n",
       "      <th>points_cum_prev_avg</th>\n",
       "      <th>points_cum_h_a</th>\n",
       "      <th>points_cum_h_a_prev</th>\n",
       "      <th>points_cum_h_a_prev_avg</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2007-09-17</td>\n",
       "      <td>Leksands IF  - AIK IF</td>\n",
       "      <td>7 - 3</td>\n",
       "      <td>3-1, 2-1, 2-1</td>\n",
       "      <td>6095.0</td>\n",
       "      <td>9826</td>\n",
       "      <td>422474</td>\n",
       "      <td>allsvenskan</td>\n",
       "      <td>2007/08</td>\n",
       "      <td>AIK</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2007-09-19</td>\n",
       "      <td>AIK IF  - Bofors IK</td>\n",
       "      <td>6 - 4</td>\n",
       "      <td>1-1, 3-1, 2-2</td>\n",
       "      <td>1554.0</td>\n",
       "      <td>9826</td>\n",
       "      <td>422481</td>\n",
       "      <td>allsvenskan</td>\n",
       "      <td>2007/08</td>\n",
       "      <td>AIK</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2007-09-21</td>\n",
       "      <td>Almtuna IS  - AIK IF</td>\n",
       "      <td>2 - 3</td>\n",
       "      <td>0-0, 1-1, 1-1, 0-1</td>\n",
       "      <td>1008.0</td>\n",
       "      <td>9826</td>\n",
       "      <td>422484</td>\n",
       "      <td>allsvenskan</td>\n",
       "      <td>2007/08</td>\n",
       "      <td>AIK</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2</td>\n",
       "      <td>5</td>\n",
       "      <td>3</td>\n",
       "      <td>1.500000</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2007-09-26</td>\n",
       "      <td>AIK IF  - IF Björklöven</td>\n",
       "      <td>6 - 2</td>\n",
       "      <td>4-0, 1-2, 1-0</td>\n",
       "      <td>1602.0</td>\n",
       "      <td>9826</td>\n",
       "      <td>422494</td>\n",
       "      <td>allsvenskan</td>\n",
       "      <td>2007/08</td>\n",
       "      <td>AIK</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3</td>\n",
       "      <td>8</td>\n",
       "      <td>5</td>\n",
       "      <td>1.666667</td>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>3.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2007-09-28</td>\n",
       "      <td>Rögle BK  - AIK IF</td>\n",
       "      <td>2 - 0</td>\n",
       "      <td>1-0, 0-0, 1-0</td>\n",
       "      <td>2988.0</td>\n",
       "      <td>9826</td>\n",
       "      <td>422499</td>\n",
       "      <td>allsvenskan</td>\n",
       "      <td>2007/08</td>\n",
       "      <td>AIK</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>8</td>\n",
       "      <td>8</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 66 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "          date                     game  score         periodscore  \\\n",
       "0  2007-09-17     Leksands IF  - AIK IF  7 - 3       3-1, 2-1, 2-1   \n",
       "1  2007-09-19       AIK IF  - Bofors IK  6 - 4       1-1, 3-1, 2-2   \n",
       "2  2007-09-21      Almtuna IS  - AIK IF  2 - 3  0-0, 1-1, 1-1, 0-1   \n",
       "3  2007-09-26   AIK IF  - IF Björklöven  6 - 2       4-0, 1-2, 1-0   \n",
       "4  2007-09-28        Rögle BK  - AIK IF  2 - 0       1-0, 0-0, 1-0   \n",
       "\n",
       "   spectators schedule_id game_id       league   season team  ... H2H_W  \\\n",
       "0      6095.0        9826  422474  allsvenskan  2007/08  AIK  ...   0.0   \n",
       "1      1554.0        9826  422481  allsvenskan  2007/08  AIK  ...   0.0   \n",
       "2      1008.0        9826  422484  allsvenskan  2007/08  AIK  ...   0.0   \n",
       "3      1602.0        9826  422494  allsvenskan  2007/08  AIK  ...   0.0   \n",
       "4      2988.0        9826  422499  allsvenskan  2007/08  AIK  ...   0.0   \n",
       "\n",
       "   H2H_D  H2H_L points points_cum points_cum_prev points_cum_prev_avg  \\\n",
       "0    0.0    0.0      0          0               0                 NaN   \n",
       "1    0.0    0.0      3          3               0            0.000000   \n",
       "2    0.0    0.0      2          5               3            1.500000   \n",
       "3    0.0    0.0      3          8               5            1.666667   \n",
       "4    0.0    0.0      0          8               8            2.000000   \n",
       "\n",
       "  points_cum_h_a  points_cum_h_a_prev  points_cum_h_a_prev_avg  \n",
       "0              0                    0                      NaN  \n",
       "1              3                    0                      NaN  \n",
       "2              2                    0                      0.0  \n",
       "3              6                    3                      3.0  \n",
       "4              2                    2                      1.0  \n",
       "\n",
       "[5 rows x 66 columns]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Create dataframe on teamlevel\n",
    "df_teams = swe.getTeamData(df_games_clean)\n",
    "df_teams.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'\\n# Read in needed coach replacement\\n#df_coachreplace = pd.read_csv(\"https://raw.githubusercontent.com/msjoelin/icehockey-analysis/master/data/shl_coach_replacement.csv\", encoding=\\'utf-8\\')\\n\\ndf_coachreplace[\\'change_date\\'] = pd.to_datetime(df_coachreplace[\\'change_date\\'], format = \\'%d/%m/%Y\\')\\n\\ndf_coachreplace.head(n=30)\\n\\ndf_teams = pd.merge(df_teams, df_coachreplace, on=[\\'season\\', \\'team\\'], how=\\'left\\')\\n\\ndf_teams[\\'date\\'] = pd.to_datetime(df_teams[\\'date\\']) \\ndf_teams.loc[(df_teams[\\'change_date\\'] > df_teams[\\'date\\']), \\'bef_after_change\\'] = \\'before\\'\\ndf_teams.loc[(df_teams[\\'change_date\\'] <= df_teams[\\'date\\']), \\'bef_after_change\\'] = \\'after\\'\\n\\n'"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\"\"\"\n",
    "# Read in needed coach replacement\n",
    "#df_coachreplace = pd.read_csv(\"https://raw.githubusercontent.com/msjoelin/icehockey-analysis/master/data/shl_coach_replacement.csv\", encoding='utf-8')\n",
    "\n",
    "df_coachreplace['change_date'] = pd.to_datetime(df_coachreplace['change_date'], format = '%d/%m/%Y')\n",
    "\n",
    "df_coachreplace.head(n=30)\n",
    "\n",
    "df_teams = pd.merge(df_teams, df_coachreplace, on=['season', 'team'], how='left')\n",
    "\n",
    "df_teams['date'] = pd.to_datetime(df_teams['date']) \n",
    "df_teams.loc[(df_teams['change_date'] > df_teams['date']), 'bef_after_change'] = 'before'\n",
    "df_teams.loc[(df_teams['change_date'] <= df_teams['date']), 'bef_after_change'] = 'after'\n",
    "\n",
    "\"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "df_teams.to_csv(\"data/df_teams.csv\", index=False)"
   ]
  }
 ],
 "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
}