plot_kde.ipynb

{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "%pip install -r requirements.txt"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {},
      "outputs": [],
      "source": [
        "from statsbombpy import sb\n",
        "import mplsoccer as mpl\n",
        "import numpy as np\n",
        "import pandas as pd\n",
        "import matplotlib.pyplot as plt\n",
        "import warnings\n",
        "warnings.filterwarnings('ignore')\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 3,
      "metadata": {},
      "outputs": [],
      "source": [
        "matches = sb.matches(competition_id=43, season_id=106)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 15,
      "metadata": {},
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "{'Argentina': 3869685, 'Denmark': 3857266, 'England': 3869354, 'Morocco': 3869552, 'Poland': 3869152, 'Australia': 3857279, 'Tunisia': 3857275}\n"
          ]
        }
      ],
      "source": [
        "french_2022 = matches[(matches['home_team'] == 'France') |( matches['away_team'] == 'France')]\n",
        "french_2022\n",
        "# Create an empty dictionary\n",
        "country_to_match_id = {}\n",
        "\n",
        "# Iterate over each row in the DataFrame\n",
        "for index, row in french_2022.iterrows():\n",
        "    # Check if the match involves France\n",
        "    if 'France' in row['home_team']:\n",
        "        # Get the other team involved in the match\n",
        "        other_team = row['away_team']\n",
        "        # Add the other team and match_id to the dictionary\n",
        "        country_to_match_id[other_team] = row['match_id']\n",
        "    else:\n",
        "        country_to_match_id[row['home_team']] = row['match_id']\n",
        "\n",
        "print(country_to_match_id)\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 11,
      "metadata": {},
      "outputs": [
        {
          "data": {
            "text/plain": [
              "Index(['ball_receipt_outcome', 'ball_recovery_recovery_failure',\n",
              "       'block_deflection', 'carry_end_location', 'clearance_aerial_won',\n",
              "       'clearance_body_part', 'clearance_head', 'clearance_left_foot',\n",
              "       'clearance_other', 'clearance_right_foot', 'counterpress',\n",
              "       'dribble_nutmeg', 'dribble_outcome', 'dribble_overrun', 'duel_outcome',\n",
              "       'duel_type', 'duration', 'foul_committed_advantage',\n",
              "       'foul_committed_card', 'foul_committed_offensive',\n",
              "       'foul_committed_type', 'foul_won_advantage', 'foul_won_defensive',\n",
              "       'goalkeeper_body_part', 'goalkeeper_end_location', 'goalkeeper_outcome',\n",
              "       'goalkeeper_position', 'goalkeeper_punched_out', 'goalkeeper_technique',\n",
              "       'goalkeeper_type', 'id', 'index', 'injury_stoppage_in_chain',\n",
              "       'interception_outcome', 'location', 'match_id', 'minute',\n",
              "       'miscontrol_aerial_won', 'off_camera', 'out', 'pass_aerial_won',\n",
              "       'pass_angle', 'pass_assisted_shot_id', 'pass_body_part', 'pass_cross',\n",
              "       'pass_cut_back', 'pass_deflected', 'pass_end_location',\n",
              "       'pass_goal_assist', 'pass_height', 'pass_inswinging', 'pass_length',\n",
              "       'pass_outcome', 'pass_outswinging', 'pass_recipient',\n",
              "       'pass_recipient_id', 'pass_shot_assist', 'pass_switch',\n",
              "       'pass_technique', 'pass_through_ball', 'pass_type', 'period',\n",
              "       'play_pattern', 'player', 'player_id', 'position', 'possession',\n",
              "       'possession_team', 'possession_team_id', 'related_events', 'second',\n",
              "       'shot_aerial_won', 'shot_body_part', 'shot_end_location',\n",
              "       'shot_first_time', 'shot_freeze_frame', 'shot_key_pass_id',\n",
              "       'shot_one_on_one', 'shot_outcome', 'shot_statsbomb_xg',\n",
              "       'shot_technique', 'shot_type', 'substitution_outcome',\n",
              "       'substitution_outcome_id', 'substitution_replacement',\n",
              "       'substitution_replacement_id', 'tactics', 'team', 'team_id',\n",
              "       'timestamp', 'type', 'under_pressure'],\n",
              "      dtype='object')"
            ]
          },
          "execution_count": 11,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "events = sb.events(match_id = '3869486')\n",
        "events.columns"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 26,
      "metadata": {},
      "outputs": [],
      "source": [
        "def visualize_heatmap(df, team):\n",
        "    events = df[~pd.isna(df['location']) & \n",
        "                    (df['team'] == \"France\")]\n",
        "    x, y = np.array(events['location'].tolist()).T\n",
        "    pitch = mpl.Pitch(line_alpha=0.5, goal_type='box',goal_alpha=0.3,pitch_color='grass', line_color='white', stripe=True)\n",
        "    fig, ax = pitch.grid(grid_height=0.9, title_height=0.06, axis=False,\n",
        "                         endnote_height=0.04, title_space=0, endnote_space=0)\n",
        "    pitch.kdeplot(x, y, cmap='Oranges', fill=True, levels=10, alpha=0.8, ax=ax['pitch'])\n",
        "    fig.suptitle(\"France's Heatmap against \"+team, fontsize = 20)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 27,
      "metadata": {},
      "outputs": [],
      "source": [
        "def shot_visualizer(events,team):\n",
        "    shots = events[(events['type'] == \"Shot\") & \n",
        "                (events['team'] == \"France\") &\n",
        "                (events['shot_type'] != \"Penalty\")]\n",
        "    coordinates = shots[['location', 'shot_end_location']]\n",
        "    x1, y1 = np.array(coordinates['location'].tolist()).T\n",
        "    x2, y2 = np.array([t[0:2] for t in  coordinates['shot_end_location'].tolist()]).T\n",
        "    shots['x1'], shots['y1'] = x1,y1\n",
        "    xg = np.array(shots['shot_statsbomb_xg'].tolist())\n",
        "    goal = ['blue' if g == \"Goal\" else 'red' for g in shots['shot_outcome'].to_list()]\n",
        "    pitch = mpl.Pitch(line_alpha=0.5,goal_type='box',goal_alpha=0.3,pitch_color='grass', line_color='white', stripe=True)\n",
        "    fig, ax = pitch.grid(grid_height=0.9, title_height=0.06, axis=False, endnote_height=0.04, title_space=0, endnote_space=0)\n",
        "    pitch.scatter(x1, y1, s=xg*500,  color=goal, linewidth=1, alpha=1, ax=ax[\"pitch\"], zorder = 3)\n",
        "    for i, row in shots.iterrows():\n",
        "        pitch.annotate(round(row.shot_statsbomb_xg,2), xy=(row.x1, row.y1), c='black', va='center', ha='center', size=12, ax=ax[\"pitch\"], zorder = 4)\n",
        "        pitch.arrows(x1, y1, x2, y2, alpha=0.4, color=goal,\n",
        "                    headaxislength=3, headlength=3, headwidth=4, width=2, ax=ax[\"pitch\"])\n",
        "    fig.suptitle(\"France's Shots against \"+team, fontsize = 20)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {},
      "outputs": [],
      "source": [
        "selected_teams = ['Argentina', 'Tunisia', 'Denmark', 'England', 'Morocco', 'Poland', 'Australia'] # replace with your actual team names\n",
        "selected_modes = ['activity', 'shotmap']\n",
        "for selected_team in selected_teams:\n",
        "    for selected_mode in selected_modes:\n",
        "        if(selected_mode == 'activity'):\n",
        "            events = sb.events(match_id = country_to_match_id[selected_team])\n",
        "            visualize_heatmap(events, selected_team)\n",
        "        else:\n",
        "            events = sb.events(match_id = country_to_match_id[selected_team])\n",
        "            shot_visualizer(events, selected_team)\n",
        "        imageName = f\"{selected_team}_{selected_mode}\"\n",
        "        imagePath = f\"data/{imageName}.png\"\n",
        "        plt.savefig(imagePath)\n"
      ]
    }
  ],
  "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.12.0"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}