docs/ratings/colley_ratings.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "c1915f05-dde7-48c3-8685-54106e8db9d3",
   "metadata": {},
   "source": [
    "# Colley Ratings"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "a23baca5-1387-4d56-9c8a-59bf2633bd84",
   "metadata": {},
   "outputs": [],
   "source": [
    "import sys\n",
    "\n",
    "sys.path.append(\"../\")\n",
    "import penaltyblog as pb"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "bea84cd8-7e75-4863-9ece-349870e19e53",
   "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></th>\n",
       "      <th></th>\n",
       "      <th>div</th>\n",
       "      <th>team_home</th>\n",
       "      <th>team_away</th>\n",
       "      <th>fthg</th>\n",
       "      <th>ftag</th>\n",
       "      <th>ftr</th>\n",
       "      <th>hthg</th>\n",
       "      <th>htag</th>\n",
       "      <th>htr</th>\n",
       "      <th>referee</th>\n",
       "      <th>...</th>\n",
       "      <th>ah_ch</th>\n",
       "      <th>b365_cahh</th>\n",
       "      <th>b365_caha</th>\n",
       "      <th>pcahh</th>\n",
       "      <th>pcaha</th>\n",
       "      <th>max_cahh</th>\n",
       "      <th>max_caha</th>\n",
       "      <th>avg_cahh</th>\n",
       "      <th>avg_caha</th>\n",
       "      <th>datetime</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>competition</th>\n",
       "      <th>season</th>\n",
       "      <th>id</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th rowspan=\"5\" valign=\"top\">ENG Premier League</th>\n",
       "      <th rowspan=\"5\" valign=\"top\">2021-2022</th>\n",
       "      <th>1628812800---brentford---arsenal</th>\n",
       "      <td>E0</td>\n",
       "      <td>Brentford</td>\n",
       "      <td>Arsenal</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>M Oliver</td>\n",
       "      <td>...</td>\n",
       "      <td>0.50</td>\n",
       "      <td>1.75</td>\n",
       "      <td>2.05</td>\n",
       "      <td>1.81</td>\n",
       "      <td>2.13</td>\n",
       "      <td>2.05</td>\n",
       "      <td>2.17</td>\n",
       "      <td>1.80</td>\n",
       "      <td>2.09</td>\n",
       "      <td>2021-08-13 20:00:00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1628899200---burnley---brighton_&amp;_hove_albion</th>\n",
       "      <td>E0</td>\n",
       "      <td>Burnley</td>\n",
       "      <td>Brighton &amp; Hove Albion</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>A</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>D Coote</td>\n",
       "      <td>...</td>\n",
       "      <td>0.25</td>\n",
       "      <td>1.79</td>\n",
       "      <td>2.15</td>\n",
       "      <td>1.81</td>\n",
       "      <td>2.14</td>\n",
       "      <td>1.82</td>\n",
       "      <td>2.19</td>\n",
       "      <td>1.79</td>\n",
       "      <td>2.12</td>\n",
       "      <td>2021-08-14 15:00:00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1628899200---chelsea---crystal_palace</th>\n",
       "      <td>E0</td>\n",
       "      <td>Chelsea</td>\n",
       "      <td>Crystal Palace</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>J Moss</td>\n",
       "      <td>...</td>\n",
       "      <td>-1.50</td>\n",
       "      <td>2.05</td>\n",
       "      <td>1.75</td>\n",
       "      <td>2.12</td>\n",
       "      <td>1.81</td>\n",
       "      <td>2.16</td>\n",
       "      <td>1.93</td>\n",
       "      <td>2.06</td>\n",
       "      <td>1.82</td>\n",
       "      <td>2021-08-14 15:00:00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1628899200---everton---southampton</th>\n",
       "      <td>E0</td>\n",
       "      <td>Everton</td>\n",
       "      <td>Southampton</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "      <td>H</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>A</td>\n",
       "      <td>A Madley</td>\n",
       "      <td>...</td>\n",
       "      <td>-0.50</td>\n",
       "      <td>2.05</td>\n",
       "      <td>1.88</td>\n",
       "      <td>2.05</td>\n",
       "      <td>1.88</td>\n",
       "      <td>2.08</td>\n",
       "      <td>1.90</td>\n",
       "      <td>2.03</td>\n",
       "      <td>1.86</td>\n",
       "      <td>2021-08-14 15:00:00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1628899200---leicester_city---wolverhampton_wanderers</th>\n",
       "      <td>E0</td>\n",
       "      <td>Leicester City</td>\n",
       "      <td>Wolverhampton Wanderers</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>H</td>\n",
       "      <td>C Pawson</td>\n",
       "      <td>...</td>\n",
       "      <td>-0.75</td>\n",
       "      <td>2.02</td>\n",
       "      <td>1.91</td>\n",
       "      <td>2.01</td>\n",
       "      <td>1.92</td>\n",
       "      <td>2.05</td>\n",
       "      <td>1.95</td>\n",
       "      <td>1.99</td>\n",
       "      <td>1.89</td>\n",
       "      <td>2021-08-14 15:00:00</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 105 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                                                div  \\\n",
       "competition        season    id                                                       \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                    E0   \n",
       "                             1628899200---burnley---brighton_&_hove_albion       E0   \n",
       "                             1628899200---chelsea---crystal_palace               E0   \n",
       "                             1628899200---everton---southampton                  E0   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  E0   \n",
       "\n",
       "                                                                                      team_home  \\\n",
       "competition        season    id                                                                   \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                         Brentford   \n",
       "                             1628899200---burnley---brighton_&_hove_albion              Burnley   \n",
       "                             1628899200---chelsea---crystal_palace                      Chelsea   \n",
       "                             1628899200---everton---southampton                         Everton   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  Leicester City   \n",
       "\n",
       "                                                                                               team_away  \\\n",
       "competition        season    id                                                                            \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                                    Arsenal   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        Brighton & Hove Albion   \n",
       "                             1628899200---chelsea---crystal_palace                        Crystal Palace   \n",
       "                             1628899200---everton---southampton                              Southampton   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  Wolverhampton Wanderers   \n",
       "\n",
       "                                                                                 fthg  \\\n",
       "competition        season    id                                                         \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                       2   \n",
       "                             1628899200---burnley---brighton_&_hove_albion          1   \n",
       "                             1628899200---chelsea---crystal_palace                  3   \n",
       "                             1628899200---everton---southampton                     3   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...     1   \n",
       "\n",
       "                                                                                 ftag  \\\n",
       "competition        season    id                                                         \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                       0   \n",
       "                             1628899200---burnley---brighton_&_hove_albion          2   \n",
       "                             1628899200---chelsea---crystal_palace                  0   \n",
       "                             1628899200---everton---southampton                     1   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...     0   \n",
       "\n",
       "                                                                                ftr  \\\n",
       "competition        season    id                                                       \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                     H   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        A   \n",
       "                             1628899200---chelsea---crystal_palace                H   \n",
       "                             1628899200---everton---southampton                   H   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...   H   \n",
       "\n",
       "                                                                                 hthg  \\\n",
       "competition        season    id                                                         \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                       1   \n",
       "                             1628899200---burnley---brighton_&_hove_albion          1   \n",
       "                             1628899200---chelsea---crystal_palace                  2   \n",
       "                             1628899200---everton---southampton                     0   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...     1   \n",
       "\n",
       "                                                                                 htag  \\\n",
       "competition        season    id                                                         \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                       0   \n",
       "                             1628899200---burnley---brighton_&_hove_albion          0   \n",
       "                             1628899200---chelsea---crystal_palace                  0   \n",
       "                             1628899200---everton---southampton                     1   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...     0   \n",
       "\n",
       "                                                                                htr  \\\n",
       "competition        season    id                                                       \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                     H   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        H   \n",
       "                             1628899200---chelsea---crystal_palace                H   \n",
       "                             1628899200---everton---southampton                   A   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...   H   \n",
       "\n",
       "                                                                                  referee  \\\n",
       "competition        season    id                                                             \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                    M Oliver   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        D Coote   \n",
       "                             1628899200---chelsea---crystal_palace                 J Moss   \n",
       "                             1628899200---everton---southampton                  A Madley   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  C Pawson   \n",
       "\n",
       "                                                                                 ...  \\\n",
       "competition        season    id                                                  ...   \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                    ...   \n",
       "                             1628899200---burnley---brighton_&_hove_albion       ...   \n",
       "                             1628899200---chelsea---crystal_palace               ...   \n",
       "                             1628899200---everton---southampton                  ...   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  ...   \n",
       "\n",
       "                                                                                 ah_ch  \\\n",
       "competition        season    id                                                          \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                     0.50   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        0.25   \n",
       "                             1628899200---chelsea---crystal_palace               -1.50   \n",
       "                             1628899200---everton---southampton                  -0.50   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...  -0.75   \n",
       "\n",
       "                                                                                 b365_cahh  \\\n",
       "competition        season    id                                                              \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                         1.75   \n",
       "                             1628899200---burnley---brighton_&_hove_albion            1.79   \n",
       "                             1628899200---chelsea---crystal_palace                    2.05   \n",
       "                             1628899200---everton---southampton                       2.05   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...       2.02   \n",
       "\n",
       "                                                                                 b365_caha  \\\n",
       "competition        season    id                                                              \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                         2.05   \n",
       "                             1628899200---burnley---brighton_&_hove_albion            2.15   \n",
       "                             1628899200---chelsea---crystal_palace                    1.75   \n",
       "                             1628899200---everton---southampton                       1.88   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...       1.91   \n",
       "\n",
       "                                                                                 pcahh  \\\n",
       "competition        season    id                                                          \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                     1.81   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        1.81   \n",
       "                             1628899200---chelsea---crystal_palace                2.12   \n",
       "                             1628899200---everton---southampton                   2.05   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...   2.01   \n",
       "\n",
       "                                                                                 pcaha  \\\n",
       "competition        season    id                                                          \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                     2.13   \n",
       "                             1628899200---burnley---brighton_&_hove_albion        2.14   \n",
       "                             1628899200---chelsea---crystal_palace                1.81   \n",
       "                             1628899200---everton---southampton                   1.88   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...   1.92   \n",
       "\n",
       "                                                                                 max_cahh  \\\n",
       "competition        season    id                                                             \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                        2.05   \n",
       "                             1628899200---burnley---brighton_&_hove_albion           1.82   \n",
       "                             1628899200---chelsea---crystal_palace                   2.16   \n",
       "                             1628899200---everton---southampton                      2.08   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...      2.05   \n",
       "\n",
       "                                                                                 max_caha  \\\n",
       "competition        season    id                                                             \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                        2.17   \n",
       "                             1628899200---burnley---brighton_&_hove_albion           2.19   \n",
       "                             1628899200---chelsea---crystal_palace                   1.93   \n",
       "                             1628899200---everton---southampton                      1.90   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...      1.95   \n",
       "\n",
       "                                                                                 avg_cahh  \\\n",
       "competition        season    id                                                             \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                        1.80   \n",
       "                             1628899200---burnley---brighton_&_hove_albion           1.79   \n",
       "                             1628899200---chelsea---crystal_palace                   2.06   \n",
       "                             1628899200---everton---southampton                      2.03   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...      1.99   \n",
       "\n",
       "                                                                                 avg_caha  \\\n",
       "competition        season    id                                                             \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                        2.09   \n",
       "                             1628899200---burnley---brighton_&_hove_albion           2.12   \n",
       "                             1628899200---chelsea---crystal_palace                   1.82   \n",
       "                             1628899200---everton---southampton                      1.86   \n",
       "                             1628899200---leicester_city---wolverhampton_wan...      1.89   \n",
       "\n",
       "                                                                                           datetime  \n",
       "competition        season    id                                                                      \n",
       "ENG Premier League 2021-2022 1628812800---brentford---arsenal                   2021-08-13 20:00:00  \n",
       "                             1628899200---burnley---brighton_&_hove_albion      2021-08-14 15:00:00  \n",
       "                             1628899200---chelsea---crystal_palace              2021-08-14 15:00:00  \n",
       "                             1628899200---everton---southampton                 2021-08-14 15:00:00  \n",
       "                             1628899200---leicester_city---wolverhampton_wan... 2021-08-14 15:00:00  \n",
       "\n",
       "[5 rows x 105 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "fbd = pb.scrapers.FootballData(\"ENG Premier League\", \"2021-2022\")\n",
    "df = fbd.get_fixtures()\n",
    "                               \n",
    "df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "6f3eb29e-a381-40bf-8d65-63f6744232f3",
   "metadata": {},
   "source": [
    "### Colley ratings including tied scorelines"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "9fc2e8b0-d55f-45c4-bb23-69cf0fdcf56c",
   "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>team</th>\n",
       "      <th>rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Liverpool</td>\n",
       "      <td>1.904762</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Manchester City</td>\n",
       "      <td>1.892857</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Chelsea</td>\n",
       "      <td>1.791667</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Tottenham Hotspur</td>\n",
       "      <td>1.708333</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Arsenal</td>\n",
       "      <td>1.672619</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Manchester United</td>\n",
       "      <td>1.654762</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Brighton &amp; Hove Albion</td>\n",
       "      <td>1.64881</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Crystal Palace</td>\n",
       "      <td>1.625</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>West Ham</td>\n",
       "      <td>1.619048</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Leicester City</td>\n",
       "      <td>1.607143</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Newcastle United</td>\n",
       "      <td>1.583333</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Wolverhampton Wanderers</td>\n",
       "      <td>1.559524</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Southampton</td>\n",
       "      <td>1.541667</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Brentford</td>\n",
       "      <td>1.529762</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Burnley</td>\n",
       "      <td>1.511905</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Aston Villa</td>\n",
       "      <td>1.511905</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Leeds United</td>\n",
       "      <td>1.505952</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>Everton</td>\n",
       "      <td>1.464286</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>Norwich City</td>\n",
       "      <td>1.339286</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Watford</td>\n",
       "      <td>1.327381</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                       team    rating\n",
       "0                 Liverpool  1.904762\n",
       "1           Manchester City  1.892857\n",
       "2                   Chelsea  1.791667\n",
       "3         Tottenham Hotspur  1.708333\n",
       "4                   Arsenal  1.672619\n",
       "5         Manchester United  1.654762\n",
       "6    Brighton & Hove Albion   1.64881\n",
       "7            Crystal Palace     1.625\n",
       "8                  West Ham  1.619048\n",
       "9            Leicester City  1.607143\n",
       "10         Newcastle United  1.583333\n",
       "11  Wolverhampton Wanderers  1.559524\n",
       "12              Southampton  1.541667\n",
       "13                Brentford  1.529762\n",
       "14                  Burnley  1.511905\n",
       "15              Aston Villa  1.511905\n",
       "16             Leeds United  1.505952\n",
       "17                  Everton  1.464286\n",
       "18             Norwich City  1.339286\n",
       "19                  Watford  1.327381"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "colley = pb.ratings.Colley(df[\"fthg\"], df[\"ftag\"], df[\"team_home\"], df[\"team_away\"])\n",
    "colley.get_ratings()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f2f7e397-82de-4086-87a4-a6e76fe1251c",
   "metadata": {},
   "source": [
    "### Colley ratings excluding tied scorelines"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "7515c4ce-e932-46f6-92da-4be85185a98f",
   "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>team</th>\n",
       "      <th>rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Liverpool</td>\n",
       "      <td>0.809524</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Manchester City</td>\n",
       "      <td>0.809524</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Chelsea</td>\n",
       "      <td>0.678571</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Tottenham Hotspur</td>\n",
       "      <td>0.630952</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Arsenal</td>\n",
       "      <td>0.607143</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Manchester United</td>\n",
       "      <td>0.547619</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>West Ham</td>\n",
       "      <td>0.52381</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Brighton &amp; Hove Albion</td>\n",
       "      <td>0.511905</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Leicester City</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Crystal Palace</td>\n",
       "      <td>0.488095</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Newcastle United</td>\n",
       "      <td>0.47619</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Wolverhampton Wanderers</td>\n",
       "      <td>0.47619</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Brentford</td>\n",
       "      <td>0.440476</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Aston Villa</td>\n",
       "      <td>0.428571</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Southampton</td>\n",
       "      <td>0.416667</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Leeds United</td>\n",
       "      <td>0.392857</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Everton</td>\n",
       "      <td>0.380952</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>Burnley</td>\n",
       "      <td>0.380952</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>Norwich City</td>\n",
       "      <td>0.25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Watford</td>\n",
       "      <td>0.25</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                       team    rating\n",
       "0                 Liverpool  0.809524\n",
       "1           Manchester City  0.809524\n",
       "2                   Chelsea  0.678571\n",
       "3         Tottenham Hotspur  0.630952\n",
       "4                   Arsenal  0.607143\n",
       "5         Manchester United  0.547619\n",
       "6                  West Ham   0.52381\n",
       "7    Brighton & Hove Albion  0.511905\n",
       "8            Leicester City       0.5\n",
       "9            Crystal Palace  0.488095\n",
       "10         Newcastle United   0.47619\n",
       "11  Wolverhampton Wanderers   0.47619\n",
       "12                Brentford  0.440476\n",
       "13              Aston Villa  0.428571\n",
       "14              Southampton  0.416667\n",
       "15             Leeds United  0.392857\n",
       "16                  Everton  0.380952\n",
       "17                  Burnley  0.380952\n",
       "18             Norwich City      0.25\n",
       "19                  Watford      0.25"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "colley = pb.ratings.Colley(df[\"fthg\"], df[\"ftag\"], df[\"team_home\"], df[\"team_away\"], include_draws=False)\n",
    "colley.get_ratings()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.10.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}