Shot_Charts/heat_map_gobert.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os, sys\n",
    "\n",
    "sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(\"__file__\"))))\n",
    "from nbafuns import *\n",
    "\n",
    "fig_DIR = \"../figs/heat_maps/\"\n",
    "data_DIR = \"../data/misc/\"\n",
    "\n",
    "bball_gray = \"#312f30\"\n",
    "bball_white = \"#dddee0\"\n",
    "bball_orange = \"#f87c24\"\n",
    "bball_light_orange = \"#fbaf7b\"\n",
    "bball_black = \"#000010\"\n",
    "dark_grey = \"#282828\"\n",
    "fontsize = 28\n",
    "title_size = 48"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Plot for Player"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "data1 = pd.read_csv(data_DIR + \"gobert_on.csv\")\n",
    "data1[\"weight\"] = 1\n",
    "data2 = pd.read_csv(data_DIR + \"gobert_off.csv\")\n",
    "data2[\"weight\"] = 0\n",
    "league = \"NBA\"\n",
    "from_year = 2023\n",
    "player = \"Rudy Gobert\"\n",
    "\n",
    "shots_df = pd.concat([data1,data2])\n",
    "shots_df[\"LOC_X\"] = -shots_df[\"LOC_X\"]\n",
    "player_id = 203497"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "fig = plt.figure(figsize=(20, 20))\n",
    "ax = fig.add_subplot(111)\n",
    "\n",
    "ax.set_xlim(252, -252)\n",
    "ax.set_ylim(-50, 423)\n",
    "ax.set_facecolor(bball_black)\n",
    "fig.set_facecolor(bball_black)\n",
    "ax.axes.get_xaxis().set_visible(False)\n",
    "ax.axes.get_yaxis().set_visible(False)\n",
    "draw_court(ax=ax, outer_lines=True, lw=3, color=bball_orange)\n",
    "sns.kdeplot(\n",
    "    x=shots_df.LOC_X, y=shots_df.LOC_Y, fill=True, cmap=\"inferno\", levels=21, ax=ax,\n",
    "    weights=shots_df.weight,\n",
    ")\n",
    "# ax.scatter(x=shots_df.LOC_X, y=shots_df.LOC_Y, facecolors=\"w\", s=5, linewidths=0.5)\n",
    "\n",
    "ax.text(\n",
    "    x=+180, y=-65, s=\"@SravanNBA\", color=bball_white, fontsize=fontsize, ha=\"center\"\n",
    ")\n",
    "ax.text(\n",
    "    x=-180,\n",
    "    y=-65,\n",
    "    s=\"Source:@daniel_bratulic\",\n",
    "    color=bball_white,\n",
    "    fontsize=fontsize,\n",
    "    ha=\"center\",\n",
    ")\n",
    "try:\n",
    "    target_dir = \"player_imgs\"\n",
    "    if f\"{player_id}.png\" not in os.listdir(target_dir):\n",
    "        if league == \"NBA\":\n",
    "            url = f\"https://cdn.nba.com/headshots/nba/latest/260x190/{player_id}.png\"\n",
    "        elif league == \"WNBA\":\n",
    "            url = f\"https://ak-static.cms.nba.com/wp-content/uploads/headshots/wnba/{player_id}.png\"\n",
    "        response = requests.get(url)\n",
    "        with open(f\"player_imgs/{player_id}.png\", \"wb\") as f:\n",
    "            f.write(response.content)\n",
    "    if league == \"NBA\":\n",
    "        zoom = 1.0\n",
    "    elif league == \"WNBA\":\n",
    "        zoom = 0.4\n",
    "    ab = AnnotationBbox(\n",
    "        OffsetImage(plt.imread(f\"player_imgs/{player_id}.png\"), zoom=zoom),\n",
    "        (-195, 380),\n",
    "        frameon=False,\n",
    "    )\n",
    "    ax.add_artist(ab)\n",
    "except:\n",
    "    pass\n",
    "if league == \"NBA\":\n",
    "    ax.text(\n",
    "        x=180,\n",
    "        y=400,\n",
    "        s=\"{0}-{1} NBA\".format(from_year, from_year + 1),\n",
    "        color=\"White\",\n",
    "        fontsize=fontsize,\n",
    "        ha=\"center\",\n",
    "        va=\"bottom\",\n",
    "    )\n",
    "elif league == \"WNBA\":\n",
    "    ax.text(\n",
    "        x=180,\n",
    "        y=400,\n",
    "        s=f\"{from_year} WNBA\",\n",
    "        color=\"White\",\n",
    "        fontsize=fontsize,\n",
    "        ha=\"center\",\n",
    "        va=\"bottom\",\n",
    "    )\n",
    "ax.text(\n",
    "    x=180,\n",
    "    y=380,\n",
    "    s=\"Regular Season\",\n",
    "    color=\"White\",\n",
    "    fontsize=fontsize,\n",
    "    ha=\"center\",\n",
    "    va=\"bottom\",\n",
    ")\n",
    "# ax.text(x=-180, y=400, s=\"Makes Only\", color='White', fontsize=fontsize, ha='center',va='bottom')\n",
    "# ax.text(x=-180, y=380, s=\"Non-Rim Attempts\", color='White', fontsize=fontsize, ha='center',va='bottom')\n",
    "ax.set_title(f\"Opposition Shot Heat Map: Gobert On vs Off\", size=title_size, color=\"Orange\")\n",
    "plt.savefig(\n",
    "    fig_DIR + f\"Shot_Heat_Map_{player}_on_off.png\",\n",
    "    bbox_inches=\"tight\",\n",
    "    facecolor=bball_black,\n",
    "    dpi=300,\n",
    ")\n",
    "plt.show()"
   ]
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "db2ffdd2fa3280b0b06d6e60ebb32d8dd74e69e88849db114a808c9d7fbf6c89"
  },
  "kernelspec": {
   "display_name": "Python 3.9.5 64-bit ('NBA-O70mvEYv': pipenv)",
   "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.11.6"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}