AllRounders.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "AllRounders.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "authorship_tag": "ABX9TyNYA5Px5kvEs46UzTEKkDnL",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/neerajcshah/all_rounders/blob/master/AllRounders.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "yx6Py0b2QfWa",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "import sys,requests,csv,io\n",
        "import pandas as pd\n",
        "from bs4 import BeautifulSoup\n",
        "from urllib.parse import urljoin\n",
        "from sklearn.preprocessing import PolynomialFeatures\n",
        "from sklearn import linear_model\n",
        "import numpy as np\n",
        "from sklearn.pipeline import make_pipeline\n",
        "import matplotlib.pyplot as plt"
      ],
      "execution_count": 93,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "q20wSAyQN5Cs",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "### Constants for scraping\n",
        "ranking_container_name = 'rankings-block__container full '\n",
        "\n",
        "rank_class_name = \"table-body__cell table-body__cell--position u-text-right\"\n",
        "rating_class_name = \"table-body__cell rating\"\n",
        "name_class_name = \"table-body__cell rankings-table__name name\"\n",
        "\n",
        "banner_class_name = \"rankings-block__banner\"\n",
        "banner_player_class_name = \"rankings-block__banner--name-large\"\n",
        "banner_ranking_class_name = \"rankings-block__position\" # Always 1\n",
        "banner_rating_class_name = \"rankings-block__banner--rating\"\n",
        "\n",
        "batting_url = \"https://www.icc-cricket.com/rankings/mens/player-rankings/test/batting\"\n",
        "bowling_url = \"https://www.icc-cricket.com/rankings/mens/player-rankings/test/bowling\"\n",
        "all_rounder_url = \"https://www.icc-cricket.com/rankings/mens/player-rankings/test/all-rounder\""
      ],
      "execution_count": 94,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "YmsyxWnyFxOW",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "### Constants for naming dataframe columns\n",
        "player = \"player\"\n",
        "batting_ranking = \"batting_ranking\"\n",
        "batting_rating = \"batting_rating\"\n",
        "bowling_ranking = \"bowling_ranking\"\n",
        "bowling_rating = \"bowling_rating\"\n",
        "all_rounder_ranking = \"all_rounder_ranking\"\n",
        "all_rounder_rating = \"all_rounder_rating\""
      ],
      "execution_count": 95,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "-3iiXT8_N7zu",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "def get_outer_data_from_tag(tag, data_tag_name, data_class_name):\n",
        "  return tag.find(name=data_tag_name, attrs={'class' : data_class_name}).find(recursive=False, text=True).strip();\n",
        "\n",
        "def get_inner_data_from_tag(tag, data_tag_name, data_class_name):\n",
        "  return tag.find(data_tag_name, {'class' : data_class_name}).text.strip();\n",
        "\n",
        "def get_dictionary_from_banner(banner_div, player_column_name, ranking_column_name, rating_column_name):\n",
        "  row_dict = {}\n",
        "  row_dict[player_column_name] = get_inner_data_from_tag(banner_div, \"div\", banner_player_class_name)\n",
        "  row_dict[ranking_column_name] = get_outer_data_from_tag(banner_div, \"td\", banner_ranking_class_name)\n",
        "  row_dict[rating_column_name] = get_outer_data_from_tag(banner_div, \"div\", banner_rating_class_name) \n",
        "  return row_dict\n",
        "\n",
        "def get_dataframe_from_ranking(ranking_div, player_column_name, ranking_column_name, rating_column_name):\n",
        "  batting_rows_list = []\n",
        "\n",
        "  # Handle single top-rated player (banner player)\n",
        "  banner_div = ranking_div.find('tr', {'class' : banner_class_name})\n",
        "  batting_rows_list.append(get_dictionary_from_banner(banner_div, player_column_name, ranking_column_name, rating_column_name))\n",
        "\n",
        "  # Handle all other players\n",
        "  batting_trs = ranking_div.find_all(\"tr\", {'class' : \"table-body\"})\n",
        "  for index, tr in enumerate(batting_trs):\n",
        "    row_dict = {}\n",
        "    row_dict[player_column_name] = get_inner_data_from_tag(tr, \"td\", name_class_name)\n",
        "    row_dict[ranking_column_name] = get_outer_data_from_tag(tr, \"td\", rank_class_name)\n",
        "    row_dict[rating_column_name] = get_outer_data_from_tag(tr, \"td\", rating_class_name) \n",
        "\n",
        "    # Handle ranking ties\n",
        "    if (row_dict[ranking_column_name] == \"\"):\n",
        "      row_dict[ranking_column_name] = index + 1\n",
        "    \n",
        "    batting_rows_list.append(row_dict)\n",
        "  \n",
        "  return pd.DataFrame(batting_rows_list, columns=[player_column_name, ranking_column_name, rating_column_name])"
      ],
      "execution_count": 96,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "J7b6LCk6Nfau",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "batting_request = requests.get(batting_url)\n",
        "bowling_request = requests.get(bowling_url)\n",
        "all_rounder_request = requests.get(all_rounder_url)\n",
        "\n",
        "batting_soup = BeautifulSoup(batting_request.content, \"lxml\")\n",
        "bowling_soup = BeautifulSoup(bowling_request.content, \"lxml\")\n",
        "all_rounder_soup = BeautifulSoup(all_rounder_request.content, \"lxml\")"
      ],
      "execution_count": 97,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "Ekd1qJBHbO2c",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "batting_outer_div = batting_soup.find('div', {'class' : ranking_container_name})\n",
        "bowling_outer_div = bowling_soup.find('div', {'class' : ranking_container_name})\n",
        "all_rounder_outer_div = all_rounder_soup.find('div', {'class' : ranking_container_name})"
      ],
      "execution_count": 98,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "z0S77KX-OKoK",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 419
        },
        "outputId": "e6f14a13-478d-49de-905b-a4a2fd5916a9"
      },
      "source": [
        "batting_df = get_dataframe_from_ranking(batting_outer_div, player, batting_ranking, batting_rating)\n",
        "batting_df"
      ],
      "execution_count": 99,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>player</th>\n",
              "      <th>batting_ranking</th>\n",
              "      <th>batting_rating</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>Steve Smith</td>\n",
              "      <td>1</td>\n",
              "      <td>911</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Virat Kohli</td>\n",
              "      <td>2</td>\n",
              "      <td>886</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Marnus Labuschagne</td>\n",
              "      <td>3</td>\n",
              "      <td>827</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>Kane Williamson</td>\n",
              "      <td>4</td>\n",
              "      <td>812</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Babar Azam</td>\n",
              "      <td>5</td>\n",
              "      <td>798</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>95</th>\n",
              "      <td>Lahiru Thirimanne</td>\n",
              "      <td>96</td>\n",
              "      <td>329</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>96</th>\n",
              "      <td>Javed Ahmadi</td>\n",
              "      <td>97</td>\n",
              "      <td>326</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>97</th>\n",
              "      <td>Marcus Harris</td>\n",
              "      <td>98</td>\n",
              "      <td>323</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>98</th>\n",
              "      <td>Theunis de Bruyn</td>\n",
              "      <td>99</td>\n",
              "      <td>317</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>99</th>\n",
              "      <td>George Linde</td>\n",
              "      <td>100</td>\n",
              "      <td>311</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>100 rows × 3 columns</p>\n",
              "</div>"
            ],
            "text/plain": [
              "                player batting_ranking batting_rating\n",
              "0          Steve Smith               1            911\n",
              "1          Virat Kohli               2            886\n",
              "2   Marnus Labuschagne               3            827\n",
              "3      Kane Williamson               4            812\n",
              "4           Babar Azam               5            798\n",
              "..                 ...             ...            ...\n",
              "95   Lahiru Thirimanne              96            329\n",
              "96        Javed Ahmadi              97            326\n",
              "97       Marcus Harris              98            323\n",
              "98    Theunis de Bruyn              99            317\n",
              "99        George Linde             100            311\n",
              "\n",
              "[100 rows x 3 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 99
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "MJSGWIRwN215",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 419
        },
        "outputId": "90c0afcc-3662-4a50-9ec1-dff3a227129d"
      },
      "source": [
        "bowling_df = get_dataframe_from_ranking(bowling_outer_div, player, bowling_ranking, bowling_rating)\n",
        "bowling_df"
      ],
      "execution_count": 100,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>player</th>\n",
              "      <th>bowling_ranking</th>\n",
              "      <th>bowling_rating</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>Pat Cummins</td>\n",
              "      <td>1</td>\n",
              "      <td>904</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Stuart Broad</td>\n",
              "      <td>2</td>\n",
              "      <td>846</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Neil Wagner</td>\n",
              "      <td>3</td>\n",
              "      <td>843</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>Tim Southee</td>\n",
              "      <td>4</td>\n",
              "      <td>812</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Jason Holder</td>\n",
              "      <td>5</td>\n",
              "      <td>810</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>95</th>\n",
              "      <td>Azhar Ali</td>\n",
              "      <td>96</td>\n",
              "      <td>63</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>96</th>\n",
              "      <td>Shan Masood</td>\n",
              "      <td>97</td>\n",
              "      <td>37</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>97</th>\n",
              "      <td>Andy McBrine</td>\n",
              "      <td>98</td>\n",
              "      <td>35</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>98</th>\n",
              "      <td>Mominul Haque</td>\n",
              "      <td>99</td>\n",
              "      <td>33</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>99</th>\n",
              "      <td>Soumya Sarkar</td>\n",
              "      <td>100</td>\n",
              "      <td>29</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>100 rows × 3 columns</p>\n",
              "</div>"
            ],
            "text/plain": [
              "           player bowling_ranking bowling_rating\n",
              "0     Pat Cummins               1            904\n",
              "1    Stuart Broad               2            846\n",
              "2     Neil Wagner               3            843\n",
              "3     Tim Southee               4            812\n",
              "4    Jason Holder               5            810\n",
              "..            ...             ...            ...\n",
              "95      Azhar Ali              96             63\n",
              "96    Shan Masood              97             37\n",
              "97   Andy McBrine              98             35\n",
              "98  Mominul Haque              99             33\n",
              "99  Soumya Sarkar             100             29\n",
              "\n",
              "[100 rows x 3 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 100
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "6jt8tfXf1RtS",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 669
        },
        "outputId": "399b6c7c-ea54-4aba-f4fd-c874469bccf0"
      },
      "source": [
        "all_rounder_df = get_dataframe_from_ranking(all_rounder_outer_div, player, all_rounder_ranking, all_rounder_rating)\n",
        "all_rounder_df"
      ],
      "execution_count": 101,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>player</th>\n",
              "      <th>all_rounder_ranking</th>\n",
              "      <th>all_rounder_rating</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>Ben Stokes</td>\n",
              "      <td>1</td>\n",
              "      <td>454</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Jason Holder</td>\n",
              "      <td>2</td>\n",
              "      <td>447</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Ravindra Jadeja</td>\n",
              "      <td>3</td>\n",
              "      <td>397</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>Mitchell Starc</td>\n",
              "      <td>4</td>\n",
              "      <td>298</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Ravichandran Ashwin</td>\n",
              "      <td>5</td>\n",
              "      <td>281</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>5</th>\n",
              "      <td>Colin de Grandhomme</td>\n",
              "      <td>6</td>\n",
              "      <td>280</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>6</th>\n",
              "      <td>Pat Cummins</td>\n",
              "      <td>7</td>\n",
              "      <td>266</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>7</th>\n",
              "      <td>Chris Woakes</td>\n",
              "      <td>7</td>\n",
              "      <td>266</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>8</th>\n",
              "      <td>Roston Chase</td>\n",
              "      <td>9</td>\n",
              "      <td>256</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>9</th>\n",
              "      <td>Stuart Broad</td>\n",
              "      <td>10</td>\n",
              "      <td>223</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>10</th>\n",
              "      <td>Tim Southee</td>\n",
              "      <td>11</td>\n",
              "      <td>211</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>11</th>\n",
              "      <td>Moeen Ali</td>\n",
              "      <td>12</td>\n",
              "      <td>188</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>12</th>\n",
              "      <td>Sam Curran</td>\n",
              "      <td>13</td>\n",
              "      <td>180</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>13</th>\n",
              "      <td>Neil Wagner</td>\n",
              "      <td>14</td>\n",
              "      <td>162</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>14</th>\n",
              "      <td>Kemar Roach</td>\n",
              "      <td>15</td>\n",
              "      <td>161</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15</th>\n",
              "      <td>Mehedi Hasan</td>\n",
              "      <td>15</td>\n",
              "      <td>161</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>16</th>\n",
              "      <td>Trent Boult</td>\n",
              "      <td>17</td>\n",
              "      <td>160</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>17</th>\n",
              "      <td>Sikandar Raza</td>\n",
              "      <td>18</td>\n",
              "      <td>157</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>18</th>\n",
              "      <td>Yasir Shah</td>\n",
              "      <td>19</td>\n",
              "      <td>154</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>19</th>\n",
              "      <td>Kagiso Rabada</td>\n",
              "      <td>20</td>\n",
              "      <td>150</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>"
            ],
            "text/plain": [
              "                 player all_rounder_ranking all_rounder_rating\n",
              "0            Ben Stokes                   1                454\n",
              "1          Jason Holder                   2                447\n",
              "2       Ravindra Jadeja                   3                397\n",
              "3        Mitchell Starc                   4                298\n",
              "4   Ravichandran Ashwin                   5                281\n",
              "5   Colin de Grandhomme                   6                280\n",
              "6           Pat Cummins                   7                266\n",
              "7          Chris Woakes                   7                266\n",
              "8          Roston Chase                   9                256\n",
              "9          Stuart Broad                  10                223\n",
              "10          Tim Southee                  11                211\n",
              "11            Moeen Ali                  12                188\n",
              "12           Sam Curran                  13                180\n",
              "13          Neil Wagner                  14                162\n",
              "14          Kemar Roach                  15                161\n",
              "15         Mehedi Hasan                  15                161\n",
              "16          Trent Boult                  17                160\n",
              "17        Sikandar Raza                  18                157\n",
              "18           Yasir Shah                  19                154\n",
              "19        Kagiso Rabada                  20                150"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 101
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "CNRiZNmcu45d",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 390
        },
        "outputId": "bbdd5877-be4a-4700-b73f-6a63fb3adc2c"
      },
      "source": [
        "complete_df = all_rounder_df.merge(bowling_df, on=player).merge(batting_df, on=player)\n",
        "complete_df"
      ],
      "execution_count": 102,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>player</th>\n",
              "      <th>all_rounder_ranking</th>\n",
              "      <th>all_rounder_rating</th>\n",
              "      <th>bowling_ranking</th>\n",
              "      <th>bowling_rating</th>\n",
              "      <th>batting_ranking</th>\n",
              "      <th>batting_rating</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>Ben Stokes</td>\n",
              "      <td>1</td>\n",
              "      <td>454</td>\n",
              "      <td>24</td>\n",
              "      <td>593</td>\n",
              "      <td>7</td>\n",
              "      <td>767</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Jason Holder</td>\n",
              "      <td>2</td>\n",
              "      <td>447</td>\n",
              "      <td>5</td>\n",
              "      <td>810</td>\n",
              "      <td>42</td>\n",
              "      <td>553</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Ravindra Jadeja</td>\n",
              "      <td>3</td>\n",
              "      <td>397</td>\n",
              "      <td>18</td>\n",
              "      <td>722</td>\n",
              "      <td>43</td>\n",
              "      <td>551</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>Mitchell Starc</td>\n",
              "      <td>4</td>\n",
              "      <td>298</td>\n",
              "      <td>7</td>\n",
              "      <td>797</td>\n",
              "      <td>86</td>\n",
              "      <td>375</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Ravichandran Ashwin</td>\n",
              "      <td>5</td>\n",
              "      <td>281</td>\n",
              "      <td>12</td>\n",
              "      <td>756</td>\n",
              "      <td>88</td>\n",
              "      <td>372</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>5</th>\n",
              "      <td>Colin de Grandhomme</td>\n",
              "      <td>6</td>\n",
              "      <td>280</td>\n",
              "      <td>32</td>\n",
              "      <td>476</td>\n",
              "      <td>31</td>\n",
              "      <td>589</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>6</th>\n",
              "      <td>Chris Woakes</td>\n",
              "      <td>7</td>\n",
              "      <td>266</td>\n",
              "      <td>19</td>\n",
              "      <td>674</td>\n",
              "      <td>79</td>\n",
              "      <td>396</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>7</th>\n",
              "      <td>Roston Chase</td>\n",
              "      <td>9</td>\n",
              "      <td>256</td>\n",
              "      <td>31</td>\n",
              "      <td>505</td>\n",
              "      <td>53</td>\n",
              "      <td>508</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>8</th>\n",
              "      <td>Moeen Ali</td>\n",
              "      <td>12</td>\n",
              "      <td>188</td>\n",
              "      <td>28</td>\n",
              "      <td>521</td>\n",
              "      <td>92</td>\n",
              "      <td>362</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>9</th>\n",
              "      <td>Sam Curran</td>\n",
              "      <td>13</td>\n",
              "      <td>180</td>\n",
              "      <td>39</td>\n",
              "      <td>407</td>\n",
              "      <td>71</td>\n",
              "      <td>444</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>10</th>\n",
              "      <td>Sikandar Raza</td>\n",
              "      <td>18</td>\n",
              "      <td>157</td>\n",
              "      <td>46</td>\n",
              "      <td>323</td>\n",
              "      <td>62</td>\n",
              "      <td>487</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>"
            ],
            "text/plain": [
              "                 player all_rounder_ranking  ... batting_ranking batting_rating\n",
              "0            Ben Stokes                   1  ...               7            767\n",
              "1          Jason Holder                   2  ...              42            553\n",
              "2       Ravindra Jadeja                   3  ...              43            551\n",
              "3        Mitchell Starc                   4  ...              86            375\n",
              "4   Ravichandran Ashwin                   5  ...              88            372\n",
              "5   Colin de Grandhomme                   6  ...              31            589\n",
              "6          Chris Woakes                   7  ...              79            396\n",
              "7          Roston Chase                   9  ...              53            508\n",
              "8             Moeen Ali                  12  ...              92            362\n",
              "9            Sam Curran                  13  ...              71            444\n",
              "10        Sikandar Raza                  18  ...              62            487\n",
              "\n",
              "[11 rows x 7 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 102
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "7RjxcrO2JQTC",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 34
        },
        "outputId": "4ee7585d-7d23-40f1-a781-c990486b4f18"
      },
      "source": [
        "### Finding the function that calculates an all-rounder's rating from his batting and bowling rating.\n",
        "X = complete_df[[bowling_rating, batting_rating]].values.astype(int)\n",
        "Y = complete_df[all_rounder_rating].to_numpy().astype(int)\n",
        "\n",
        "poly = PolynomialFeatures(interaction_only=True)\n",
        "\n",
        "# Transform [x1, x2] to [1, x1, x2, x1 * x2]\n",
        "X_ = poly.fit_transform(X)\n",
        "\n",
        "\n",
        "clf = linear_model.LinearRegression()\n",
        "clf.fit(X_, Y)\n",
        "\n",
        "print(\"The coefficients a, b, c, d in <a * 1 + b * x1 + c * x2 + d * x1 * x2> are:\", clf.coef_)\n",
        "\n",
        "# Optionally predict values with these coefficients.\n",
        "# clf.predict(X_)"
      ],
      "execution_count": 103,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "The coefficients a, b, c, d in <a * 1 + b * x1 + c * x2 + d * x1 * x2> are: [0.         0.0034911  0.00516932 0.00099088]\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "10yh-UvRzCoz",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        ""
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}