source/examples/demo/wind_rose.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "0b211ca8-bd70-46fc-b7b1-7c7e6bdcfe5e",
   "metadata": {},
   "source": [
    "# Wind Rose\n",
    "\n",
    "The notebook is inspired by [this plot](https://mesonet.cdn.columbiascanner.org/onsite/windrose/IA_ASOS/PEA/PEA_yearly.png)."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "a5ae369f-2486-4e7c-81c4-79ed555154d5",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:55.266735Z",
     "iopub.status.busy": "2024-07-05T16:56:55.266509Z",
     "iopub.status.idle": "2024-07-05T16:56:55.584981Z",
     "shell.execute_reply": "2024-07-05T16:56:55.584564Z"
    }
   },
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "\n",
    "from lets_plot import *\n",
    "from lets_plot.mapping import as_discrete"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "bfdcdd13-23ea-4a38-be8f-561192a555db",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:55.586681Z",
     "iopub.status.busy": "2024-07-05T16:56:55.586493Z",
     "iopub.status.idle": "2024-07-05T16:56:55.589225Z",
     "shell.execute_reply": "2024-07-05T16:56:55.589024Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "            <div id=\"nrrjQO\"></div>\n",
       "            <script type=\"text/javascript\" data-lets-plot-script=\"library\">\n",
       "                if(!window.letsPlotCallQueue) {\n",
       "                    window.letsPlotCallQueue = [];\n",
       "                }; \n",
       "                window.letsPlotCall = function(f) {\n",
       "                    window.letsPlotCallQueue.push(f);\n",
       "                };\n",
       "                (function() {\n",
       "                    var script = document.createElement(\"script\");\n",
       "                    script.type = \"text/javascript\";\n",
       "                    script.src = \"https://cdn.jsdelivr.net/gh/JetBrains/lets-plot@v4.3.3/js-package/distr/lets-plot.min.js\";\n",
       "                    script.onload = function() {\n",
       "                        window.letsPlotCall = function(f) {f();};\n",
       "                        window.letsPlotCallQueue.forEach(function(f) {f();});\n",
       "                        window.letsPlotCallQueue = [];\n",
       "                        \n",
       "                    };\n",
       "                    script.onerror = function(event) {\n",
       "                        window.letsPlotCall = function(f) {};    // noop\n",
       "                        window.letsPlotCallQueue = [];\n",
       "                        var div = document.createElement(\"div\");\n",
       "                        div.style.color = 'darkred';\n",
       "                        div.textContent = 'Error loading Lets-Plot JS';\n",
       "                        document.getElementById(\"nrrjQO\").appendChild(div);\n",
       "                    };\n",
       "                    var e = document.getElementById(\"nrrjQO\");\n",
       "                    e.appendChild(script);\n",
       "                })()\n",
       "            </script>\n",
       "            "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "LetsPlot.setup_html()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "ccdb0b79-81d3-4d64-a208-0cd76765df50",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:55.590275Z",
     "iopub.status.busy": "2024-07-05T16:56:55.590203Z",
     "iopub.status.idle": "2024-07-05T16:56:59.056506Z",
     "shell.execute_reply": "2024-07-05T16:56:59.056146Z"
    }
   },
   "outputs": [],
   "source": [
    "wind_df = pd.read_csv(\"https://raw.githubusercontent.com/JetBrains/lets-plot-docs/master/data/pea_wind.csv\")\n",
    "records_count = wind_df.shape[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "76ab511d-673c-40ba-abc0-12b88db98f5e",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:59.058128Z",
     "iopub.status.busy": "2024-07-05T16:56:59.058031Z",
     "iopub.status.idle": "2024-07-05T16:56:59.524746Z",
     "shell.execute_reply": "2024-07-05T16:56:59.524397Z"
    }
   },
   "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>station</th>\n",
       "      <th>valid</th>\n",
       "      <th>drct</th>\n",
       "      <th>sped</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>PEA</td>\n",
       "      <td>2002-05-17 00:14</td>\n",
       "      <td>330.0</td>\n",
       "      <td>8.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>PEA</td>\n",
       "      <td>2002-05-17 00:20</td>\n",
       "      <td>330.0</td>\n",
       "      <td>8.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>PEA</td>\n",
       "      <td>2002-05-17 00:34</td>\n",
       "      <td>340.0</td>\n",
       "      <td>10.35</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>PEA</td>\n",
       "      <td>2002-05-17 00:40</td>\n",
       "      <td>340.0</td>\n",
       "      <td>10.35</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>PEA</td>\n",
       "      <td>2002-05-17 00:54</td>\n",
       "      <td>340.0</td>\n",
       "      <td>14.95</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  station             valid   drct   sped\n",
       "0     PEA  2002-05-17 00:14  330.0   8.05\n",
       "1     PEA  2002-05-17 00:20  330.0   8.05\n",
       "2     PEA  2002-05-17 00:34  340.0  10.35\n",
       "3     PEA  2002-05-17 00:40  340.0  10.35\n",
       "4     PEA  2002-05-17 00:54  340.0  14.95"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def is_float(x):\n",
    "    try:\n",
    "        float(x)\n",
    "    except ValueError:\n",
    "        return False\n",
    "    return True\n",
    "\n",
    "\n",
    "wind_df = wind_df[wind_df['sped'].apply(lambda x: is_float(x))]\n",
    "wind_df = wind_df[wind_df['drct'].apply(lambda x: is_float(x))]\n",
    "\n",
    "wind_df['sped'] = wind_df['sped'].astype(float)\n",
    "wind_df['drct'] = wind_df['drct'].astype(float)\n",
    "\n",
    "wind_df = wind_df[wind_df['sped'].apply(lambda x: x >= 2.0)]\n",
    "\n",
    "wind_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "a6731d90-ab72-466a-bd96-4f01f819bec5",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:59.526118Z",
     "iopub.status.busy": "2024-07-05T16:56:59.526004Z",
     "iopub.status.idle": "2024-07-05T16:56:59.553256Z",
     "shell.execute_reply": "2024-07-05T16:56:59.552897Z"
    }
   },
   "outputs": [],
   "source": [
    "# Compute calm\n",
    "calm = 100 - wind_df.shape[0] / records_count * 100\n",
    "\n",
    "# Define the speed bins\n",
    "bins = [2, 5, 7, 10, 15, 20, float('inf')]\n",
    "bin_ids = list(range(6))\n",
    "\n",
    "wind_df['speed_group'] = pd.cut(wind_df['sped'], bins=bins, labels=bin_ids, right=False)\n",
    "\n",
    "# Group by 'drct' and 'speed_group', and count the occurrences\n",
    "grouped_counts = wind_df.groupby(['drct', 'speed_group'], observed=False).size().reset_index(name='count')\n",
    "\n",
    "# Calculate the total number of observations in the dataset\n",
    "total_observations = wind_df.shape[0]\n",
    "\n",
    "# Calculate the percentage of each speed group within each direction relative to the total number of observations\n",
    "grouped_counts['percentage_of_total'] = (grouped_counts['count'] / total_observations) * 100"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "51befd5b-c99e-40a0-b968-f87d94644fab",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2024-07-05T16:56:59.554892Z",
     "iopub.status.busy": "2024-07-05T16:56:59.554798Z",
     "iopub.status.idle": "2024-07-05T16:56:59.585834Z",
     "shell.execute_reply": "2024-07-05T16:56:59.585543Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "   <div id=\"LL0Ysf\"></div>\n",
       "   <script type=\"text/javascript\" data-lets-plot-script=\"plot\">\n",
       "       (function() {\n",
       "           var plotSpec={\n",
       "\"data\":{\n",
       "\"drct\":[10.0,10.0,10.0,10.0,10.0,10.0,20.0,20.0,20.0,20.0,20.0,20.0,30.0,30.0,30.0,30.0,30.0,30.0,40.0,40.0,40.0,40.0,40.0,40.0,50.0,50.0,50.0,50.0,50.0,50.0,60.0,60.0,60.0,60.0,60.0,60.0,70.0,70.0,70.0,70.0,70.0,70.0,80.0,80.0,80.0,80.0,80.0,80.0,90.0,90.0,90.0,90.0,90.0,90.0,100.0,100.0,100.0,100.0,100.0,100.0,110.0,110.0,110.0,110.0,110.0,110.0,120.0,120.0,120.0,120.0,120.0,120.0,130.0,130.0,130.0,130.0,130.0,130.0,140.0,140.0,140.0,140.0,140.0,140.0,150.0,150.0,150.0,150.0,150.0,150.0,160.0,160.0,160.0,160.0,160.0,160.0,170.0,170.0,170.0,170.0,170.0,170.0,180.0,180.0,180.0,180.0,180.0,180.0,190.0,190.0,190.0,190.0,190.0,190.0,200.0,200.0,200.0,200.0,200.0,200.0,210.0,210.0,210.0,210.0,210.0,210.0,220.0,220.0,220.0,220.0,220.0,220.0,230.0,230.0,230.0,230.0,230.0,230.0,240.0,240.0,240.0,240.0,240.0,240.0,250.0,250.0,250.0,250.0,250.0,250.0,260.0,260.0,260.0,260.0,260.0,260.0,270.0,270.0,270.0,270.0,270.0,270.0,280.0,280.0,280.0,280.0,280.0,280.0,290.0,290.0,290.0,290.0,290.0,290.0,300.0,300.0,300.0,300.0,300.0,300.0,310.0,310.0,310.0,310.0,310.0,310.0,320.0,320.0,320.0,320.0,320.0,320.0,330.0,330.0,330.0,330.0,330.0,330.0,340.0,340.0,340.0,340.0,340.0,340.0,350.0,350.0,350.0,350.0,350.0,350.0,360.0,360.0,360.0,360.0,360.0,360.0],\n",
       "\"speed_group\":[0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0,0.0,1.0,2.0,3.0,4.0,5.0],\n",
       "\"percentage_of_total\":[0.5965102522656505,0.4400585524047074,0.3258223632344609,0.3340620171283077,0.062153438631362974,0.010477584581064462,0.5673154292096747,0.37149649407813023,0.2370172046007786,0.23467754979141467,0.0353999945068974,0.008341378015993066,0.5413757780623794,0.35735684109980054,0.22867582658478552,0.22094479330167002,0.03438275328543484,0.00722241267238424,0.4857326832483767,0.30822409010315843,0.1560448033723581,0.1170844645903417,0.009155170993163122,0.0015258618321938536,0.4170689007996533,0.2196223797137687,0.09389136474099513,0.0434362001564517,0.0012206894657550828,2.034482442925138E-4,0.46986372019356065,0.26458444170241424,0.11108274138371253,0.07853102229691034,0.007832757405261782,5.086206107312845E-4,0.5644671537895796,0.4002844206455209,0.2020241065824662,0.15278963146367788,0.022175858627884005,0.004170689007996533,0.6440154173079524,0.5069930247769444,0.3290775351431411,0.2933723682698049,0.036010339239774945,0.008341378015993066,0.6419809348650274,0.534153365389995,0.36529132262720854,0.3211430536157331,0.03621378748406746,0.0032551719086802208,0.794668842206559,0.6189912832599733,0.42866545072432666,0.37261545942173907,0.04302930366786667,0.0068155161837992135,1.2473411857574022,1.0193774280276406,0.6783981705933874,0.6424895554757586,0.10396205283347455,0.028177581834513164,1.32485496683285,1.2287256714046373,0.8278309060262387,0.9216205466450876,0.17669480016804825,0.04770861328659449,1.1301549970449143,1.1293412040677442,0.8770653811450271,0.9836722611543043,0.17883100673311964,0.03570516687333618,1.0685101790242826,1.1418532710917337,0.8434964208367624,0.8846946903059963,0.1482120459670963,0.018005169619887473,0.9714653664967534,1.1221187913953599,0.8957826196199383,0.9759412278711888,0.1539085968072867,0.020853445039982668,0.8215240104531708,1.0114429465002324,0.8482774545776364,0.9810274339785017,0.19297065971144936,0.04770861328659449,0.7422809193012366,0.851329178242024,0.674329205707537,0.9866222606965458,0.26682237238963186,0.09592584718392026,0.6292654195967452,0.6701585166995405,0.5507343972998349,0.9384050267992199,0.26855168246611827,0.09256895115309378,0.6545947260111632,0.6408619695214186,0.5761654278363991,0.9266050286302542,0.2540051329992035,0.08321033191563815,0.610548181121834,0.5902033566925826,0.5068913006547981,0.7640498814405356,0.19775169345232343,0.05411723298180867,0.5885757707382424,0.5414775021845255,0.4777982017209687,0.6552050707440408,0.16509825024337496,0.0434362001564517,0.5325257794356549,0.5162499198922538,0.3989620070576196,0.5517516385212975,0.13142756581296391,0.03489137389616612,0.5002792327152915,0.45185855057367313,0.3564413240004842,0.45440165362732965,0.08738102092363469,0.026956892368758083,0.5413757780623794,0.5176740576023015,0.3922482149959667,0.4445344137791427,0.07466550565535257,0.02237930687217652,0.5599912924151443,0.5789119791343481,0.4089309710279528,0.3961137316375244,0.0781241258083253,0.03316206381967975,0.5374085372986752,0.6445240379186837,0.5168602646251313,0.580844737455127,0.13569997894310673,0.06886723069301592,0.49651544019588,0.5751481866149366,0.5099430243191858,0.7366860925831925,0.21067065696489806,0.10986205191795746,0.4378206217174897,0.5476826736354472,0.4687447548499518,0.7909050496871475,0.31788788170705284,0.16957411161781027,0.4875637174470093,0.5890843913489737,0.5524637073763212,1.027922254287926,0.47037234080429197,0.26377064872524414,0.5098413001970397,0.6265188682987963,0.677482653494071,1.403691161496199,0.6302826608182077,0.39641890400396324,0.5548033621856852,0.7358722996060224,0.7975171176266541,1.663799741824178,0.7029136840306353,0.4067947644628814,0.5127912997392811,0.7268188527350056,0.8001619448024568,1.6351135393789333,0.7783929826631578,0.44667062034421406,0.5039413011125567,0.6574430014312584,0.6443205896743912,1.1178463782652173,0.49427750950866234,0.26844995834397195,0.604241285548766,0.7059654076950229,0.6235688687565548,1.0018808790184843,0.30344305636228436,0.1284775662707225,0.6898929963959143,0.6288585231081602,0.5067895765326519,0.7156291992989174,0.17109997345000413,0.053913784737516166,0.7376016096825089,0.5689430151640149,0.4003861447676672,0.47179647851433953,0.08514309023641703,0.015360342444084794]\n",
       "},\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "\"series_annotations\":[{\n",
       "\"column\":\"speed_group\",\n",
       "\"factor_levels\":[0.0,1.0,2.0,3.0,4.0,5.0]\n",
       "}]\n",
       "},\n",
       "\"ggtitle\":{\n",
       "\"text\":\"Wind rose for [PEA] PELLA\",\n",
       "\"subtitle\":\"Observations from 2002 to 2024\"\n",
       "},\n",
       "\"caption\":{\n",
       "\"text\":\"Data is provided by the Iowa Environmental Mesonet of Iowa State University\"\n",
       "},\n",
       "\"ggsize\":{\n",
       "\"width\":800.0,\n",
       "\"height\":800.0\n",
       "},\n",
       "\"theme\":{\n",
       "\"name\":\"minimal2\",\n",
       "\"axis_title\":{\n",
       "\"blank\":true\n",
       "},\n",
       "\"axis_text_x\":{\n",
       "\"size\":18.0,\n",
       "\"blank\":false\n",
       "},\n",
       "\"axis_ticks_y\":{\n",
       "\"blank\":false\n",
       "},\n",
       "\"panel_grid\":{\n",
       "\"color\":\"#A0A0A0\",\n",
       "\"blank\":false\n",
       "},\n",
       "\"panel_grid_minor_x\":{\n",
       "\"blank\":false\n",
       "},\n",
       "\"plot_title\":{\n",
       "\"face\":\"bold\",\n",
       "\"size\":24.0,\n",
       "\"blank\":false\n",
       "},\n",
       "\"plot_subtitle\":{\n",
       "\"size\":18.0,\n",
       "\"blank\":false\n",
       "}\n",
       "},\n",
       "\"coord\":{\n",
       "\"name\":\"polar\",\n",
       "\"ylim\":[-1.0,null],\n",
       "\"start\":0.08722222222222223\n",
       "},\n",
       "\"kind\":\"plot\",\n",
       "\"scales\":[{\n",
       "\"name\":\"Wind Speed (mph):\",\n",
       "\"aesthetic\":\"fill\",\n",
       "\"breaks\":[0.0,1.0,2.0,3.0,4.0,5.0],\n",
       "\"labels\":[\"2 - 4.9\",\"5 - 6.9\",\"7 - 9.9\",\"10 - 14.9\",\"15 - 19.9\",\"20+\"],\n",
       "\"values\":[\"#002bff\",\"#03d3f8\",\"#7afe81\",\"#fde609\",\"#ff4404\",\"#780200\"]\n",
       "},{\n",
       "\"aesthetic\":\"y\",\n",
       "\"breaks\":[0.0,1.0,2.0,3.0,4.0,5.0],\n",
       "\"format\":\"{}%\"\n",
       "},{\n",
       "\"aesthetic\":\"x\",\n",
       "\"breaks\":[45.0,90.0,135.0,180.0,225.0,270.0,315.0,360.0],\n",
       "\"labels\":[\"NE\",\"E\",\"SE\",\"S\",\"SW\",\"W\",\"NW\",\"N\"]\n",
       "}],\n",
       "\"layers\":[{\n",
       "\"geom\":\"bar\",\n",
       "\"stat\":\"identity\",\n",
       "\"mapping\":{\n",
       "\"x\":\"drct\",\n",
       "\"y\":\"percentage_of_total\",\n",
       "\"fill\":\"speed_group\"\n",
       "},\n",
       "\"tooltips\":{\n",
       "\"formats\":[{\n",
       "\"field\":\"^y\",\n",
       "\"format\":\"{.2g}%\"\n",
       "},{\n",
       "\"field\":\"^x\",\n",
       "\"format\":\"{}°\"\n",
       "}]\n",
       "},\n",
       "\"data_meta\":{\n",
       "\"mapping_annotations\":[{\n",
       "\"aes\":\"fill\",\n",
       "\"annotation\":\"as_discrete\",\n",
       "\"parameters\":{\n",
       "\"label\":\"speed_group\",\n",
       "\"order\":1.0\n",
       "}\n",
       "}],\n",
       "\"series_annotations\":[{\n",
       "\"column\":\"speed_group\",\n",
       "\"factor_levels\":[0.0,1.0,2.0,3.0,4.0,5.0]\n",
       "}]\n",
       "},\n",
       "\"size\":0.0,\n",
       "\"width\":0.8,\n",
       "\"data\":{\n",
       "}\n",
       "},{\n",
       "\"geom\":\"rect\",\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"xmin\":5.0,\n",
       "\"xmax\":365.0,\n",
       "\"ymin\":-1.0,\n",
       "\"ymax\":0.0,\n",
       "\"fill\":\"white\",\n",
       "\"size\":0.0,\n",
       "\"data\":{\n",
       "}\n",
       "},{\n",
       "\"geom\":\"hline\",\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"yintercept\":0.0,\n",
       "\"size\":2.0,\n",
       "\"data\":{\n",
       "}\n",
       "},{\n",
       "\"geom\":\"text\",\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"x\":180.0,\n",
       "\"y\":-1.0,\n",
       "\"label\":\"Calm\\n14.3%\",\n",
       "\"hjust\":\"middle\",\n",
       "\"vjust\":\"center\",\n",
       "\"size\":\"10\",\n",
       "\"data\":{\n",
       "}\n",
       "}],\n",
       "\"metainfo_list\":[]\n",
       "};\n",
       "           var plotContainer = document.getElementById(\"LL0Ysf\");\n",
       "           window.letsPlotCall(function() {{\n",
       "               LetsPlot.buildPlotFromProcessedSpecs(plotSpec, -1, -1, plotContainer);\n",
       "           }});\n",
       "       })();\n",
       "   </script>"
      ],
      "text/plain": [
       "<lets_plot.plot.core.PlotSpec at 0x762e8ad2d0c0>"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ggplot(grouped_counts) + \\\n",
    "    geom_bar(\n",
    "        aes('drct', 'percentage_of_total', fill=as_discrete('speed_group', order=1)), \n",
    "        size=0, width=.8,\n",
    "        stat='identity',\n",
    "        tooltips=layer_tooltips().format('^y', '{.2g}%').format('^x', '{}°')\n",
    "    ) + \\\n",
    "    geom_rect(\n",
    "        # Visually align the width of the rectangle with the bars - widen it by 5 (half a bar width)\n",
    "        xmin=5, xmax=365, \n",
    "        ymin=-1, ymax=0, fill='white', size=0\n",
    "    ) + \\\n",
    "    geom_hline(yintercept=0, size=2) + \\\n",
    "    geom_text(x=180, y=-1, label=f'Calm\\n{calm:.1f}%', hjust='middle', vjust='center', size='10') + \\\n",
    "    scale_fill_manual(\n",
    "        name='Wind Speed (mph):', \n",
    "        values=['#002bff', '#03d3f8', '#7afe81', '#fde609', '#ff4404', '#780200'], \n",
    "        labels={\n",
    "            0: '2 - 4.9', \n",
    "            1: '5 - 6.9', \n",
    "            2: '7 - 9.9', \n",
    "            3: '10 - 14.9', \n",
    "            4: '15 - 19.9', \n",
    "            5: '20+'\n",
    "        },\n",
    "    ) + \\\n",
    "    scale_y_continuous(\n",
    "        breaks=[0, 1, 2, 3, 4, 5], # To not add automatically generated ticks for values outside of the data range\n",
    "        format='{}%'\n",
    "    ) + \\\n",
    "    scale_x_continuous(\n",
    "        labels={\n",
    "            45:  'NE', \n",
    "            90:  'E', \n",
    "            135: 'SE', \n",
    "            180: 'S', \n",
    "            225: 'SW',\n",
    "            270: 'W',\n",
    "            315: 'NW',\n",
    "            360: 'N', \n",
    "        },\n",
    "    ) + \\\n",
    "    labs(\n",
    "        title=\"Wind rose for [PEA] PELLA\",\n",
    "        subtitle=\"Observations from 2002 to 2024\",\n",
    "        caption=\"Data is provided by the Iowa Environmental Mesonet of Iowa State University\"\n",
    "    ) + \\\n",
    "    ggsize(800, 800) + \\\n",
    "    theme_minimal2() + \\\n",
    "    theme(\n",
    "        plot_title=element_text(size=24, face='bold'),\n",
    "        plot_subtitle=element_text(size=18),\n",
    "        panel_grid_minor_x=element_line(),\n",
    "        panel_grid=element_line(color='#A0A0A0'),\n",
    "        axis_ticks_y=element_line(),\n",
    "        axis_text_x=element_text(size=18),\n",
    "        axis_title=element_blank()\n",
    "    ) + \\\n",
    "    coord_polar(\n",
    "        ylim=[-1, None], # -1 is to make inner circle\n",
    "        start=(3.14 * 2) / 36 / 2 # Divide by 2 (i.e. rotate by half a bar width) to make the N-S axis perpendicular\n",
    "    )"
   ]
  }
 ],
 "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.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}