docs/f-24a/geom_curve_on_map.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "incomplete-caribbean",
   "metadata": {},
   "source": [
    "# `geom_curve()` on Map\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "bibliographic-beijing",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "from lets_plot import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "controversial-square",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "            <div id=\"Jqkxxz\"></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.0rc2/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(\"Jqkxxz\").appendChild(div);\n",
       "                    };\n",
       "                    var e = document.getElementById(\"Jqkxxz\");\n",
       "                    e.appendChild(script);\n",
       "                })()\n",
       "            </script>\n",
       "            "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "LetsPlot.setup_html()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "local-fifth",
   "metadata": {},
   "outputs": [],
   "source": [
    "def set_stay_time(data):\n",
    "    current_moves = moves_df[(moves_df.departure == data['name'])|(moves_df.arrival == data['name'])]\n",
    "    years = current_moves.year.sort_values().unique()\n",
    "    years_combined = []\n",
    "    for year in years:\n",
    "        if len(years_combined) == 0 or years_combined[-1][1] + 1 != year:\n",
    "            years_combined.append((year, year))\n",
    "        else:\n",
    "            years_combined[-1] = (years_combined[-1][0], year)\n",
    "    years_combined = [str(year_from) if year_from == year_to else '{0}-{1}'.format(year_from, year_to) \\\n",
    "                      for year_from, year_to in years_combined]\n",
    "    data['years'] = ', '.join(years_combined)\n",
    "    return data\n",
    "\n",
    "places_df = pd.read_csv('https://raw.githubusercontent.com/JetBrains/lets-plot-docs/master/data/pushkin/places.csv')\n",
    "moves_df = pd.read_csv('https://raw.githubusercontent.com/JetBrains/lets-plot-docs/master/data/pushkin/moves.csv')\n",
    "df = moves_df.merge(places_df, left_on='departure', right_on='name')\\\n",
    "             .rename(columns=dict(longitude='from_lon', latitude='from_lat'))\\\n",
    "             .drop(columns=['name'])\\\n",
    "             .merge(places_df, left_on='arrival', right_on='name')\\\n",
    "             .rename(columns=dict(longitude='to_lon', latitude='to_lat'))\\\n",
    "             .drop(columns=['name'])\n",
    "places_df['years'] = ''\n",
    "places_df = places_df.apply(set_stay_time, axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "sitting-shift",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "   <div id=\"DLbHX4\"></div>\n",
       "   <script type=\"text/javascript\" data-lets-plot-script=\"plot\">\n",
       "       (function() {\n",
       "           var plotSpec={\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"coord\":{\n",
       "\"name\":\"cartesian\",\n",
       "\"xlim\":[26.0,58.0],\n",
       "\"ylim\":[38.0,62.0],\n",
       "\"flip\":false\n",
       "},\n",
       "\"ggtitle\":{\n",
       "\"text\":\"Alexander Pushkin's Trips\"\n",
       "},\n",
       "\"ggsize\":{\n",
       "\"width\":1000.0,\n",
       "\"height\":800.0\n",
       "},\n",
       "\"theme\":{\n",
       "\"name\":\"classic\",\n",
       "\"line\":\"blank\",\n",
       "\"axis\":\"blank\",\n",
       "\"legend_title\":\"blank\",\n",
       "\"legend_position\":[1.0,1.0],\n",
       "\"legend_justification\":[1.0,1.0]\n",
       "},\n",
       "\"kind\":\"plot\",\n",
       "\"scales\":[{\n",
       "\"name\":\"trip name\",\n",
       "\"aesthetic\":\"color\",\n",
       "\"values\":[\"#addd8e\",\"#e34a33\",\"#8856a7\",\"#2c7fb8\",\"#1c9099\",\"#006d2c\",\"#fec44f\",\"#636363\"]\n",
       "}],\n",
       "\"layers\":[{\n",
       "\"geom\":\"livemap\",\n",
       "\"mapping\":{\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"tiles\":{\n",
       "\"kind\":\"vector_lets_plot\",\n",
       "\"url\":\"wss://tiles.datalore.jetbrains.com\",\n",
       "\"theme\":\"color\",\n",
       "\"attribution\":\"Map: <a href=\\\"https://github.com/JetBrains/lets-plot\\\">© Lets-Plot</a>, map data: <a href=\\\"https://www.openstreetmap.org/copyright\\\">© OpenStreetMap contributors</a>.\"\n",
       "},\n",
       "\"geocoding\":{\n",
       "\"url\":\"https://geo2.datalore.jetbrains.com/map_data/geocoding\"\n",
       "},\n",
       "\"const_size_zoomin\":3.0,\n",
       "\"data\":{\n",
       "}\n",
       "},{\n",
       "\"geom\":\"curve\",\n",
       "\"data\":{\n",
       "\"path\":[\"To Lyceum\",\"Regular routes\",\"Regular routes\",\"Regular routes\",\"Travel to the Caucasus\",\"Autumn at Boldino\",\"Trip researching the Pugachev rebellion\",\"Trip researching the Pugachev rebellion\",\"Regular routes\",\"Regular routes\",\"Southern exile\",\"Mikhailovsky exile\",\"Final journey\",\"Regular routes\",\"Regular routes\",\"Mikhailovsky exile\",\"Autumn at Boldino\",\"Trip researching the Pugachev rebellion\",\"Regular routes\",\"Regular routes\",\"Southern exile\",\"Regular routes\",\"Mikhailovsky exile\",\"Southern exile\",\"Regular routes\",\"Southern exile\",\"Regular routes\",\"Southern exile\",\"Travel to the Caucasus\",\"Southern exile\",\"Travel to the Caucasus\",\"Southern exile\",\"Southern exile\",\"Travel to the Caucasus\",\"Southern exile\",\"Southern exile\",\"Travel to the Caucasus\",\"Southern exile\",\"Southern exile\",\"Southern exile\",\"Regular routes\",\"Mikhailovsky exile\",\"Mikhailovsky exile\",\"Travel to the Caucasus\",\"Travel to the Caucasus\",\"Travel to the Caucasus\",\"Trip researching the Pugachev rebellion\",\"Trip researching the Pugachev rebellion\",\"Trip researching the Pugachev rebellion\",\"Trip researching the Pugachev rebellion\"],\n",
       "\"from_lat\":[55.751244,55.751244,55.751244,55.751244,55.751244,55.751244,51.23333,55.751244,55.751244,55.751244,59.93863,51.50551,59.93863,59.93863,59.93863,57.8136,55.004477,55.004477,54.728958,54.728958,55.187222,47.00367,55.187222,50.431782,50.431782,46.469391,46.469391,48.450001,52.929697,47.414101,47.414101,44.04861,45.0428,45.0428,45.0428,44.1497667,44.1497667,45.04484,44.957813,47.00367,47.00367,46.469391,57.061332,43.03667,41.716667,40.60199,56.32753,55.78874,54.32824,51.776272],\n",
       "\"from_lon\":[37.618423,37.618423,37.618423,37.618423,37.618423,37.618423,51.36667,37.618423,37.618423,37.618423,30.31413,31.28487,30.31413,30.31413,30.31413,28.3496,45.30885,45.30885,35.971598,35.971598,30.205116,28.907089,30.205116,30.516382,30.516382,30.740883,30.740883,34.983334,36.098689,40.110401,40.110401,43.05944,41.9734,41.9734,41.9734,43.4577689,43.4577689,38.97603,34.109547,28.907089,28.907089,30.740883,28.919346,44.66778,44.783333,43.09495,44.000717,49.12214,48.38657,55.099594],\n",
       "\"to_lat\":[59.93863,59.93863,59.93863,59.93863,52.929697,55.004477,55.004477,56.32753,54.728958,54.728958,55.187222,55.187222,57.022162,55.751244,55.751244,55.751244,55.751244,55.751244,55.751244,55.751244,50.431782,50.431782,57.061332,48.450001,47.00367,47.00367,47.00367,47.414101,47.414101,45.0428,45.0428,45.0428,44.1497667,44.1497667,45.04484,44.04861,43.03667,44.957813,46.469391,46.469391,46.469391,51.50551,57.8136,41.716667,40.60199,39.90861,55.78874,54.32824,51.776272,51.23333],\n",
       "\"to_lon\":[30.31413,30.31413,30.31413,30.31413,36.098689,45.30885,45.30885,44.000717,35.971598,35.971598,30.205116,30.205116,28.920945,37.618423,37.618423,37.618423,37.618423,37.618423,37.618423,37.618423,30.516382,30.516382,28.919346,34.983334,28.907089,28.907089,28.907089,40.110401,40.110401,41.9734,41.9734,41.9734,43.4577689,43.4577689,38.97603,43.05944,44.66778,34.109547,30.740883,30.740883,30.740883,31.28487,28.3496,44.783333,43.09495,41.27694,49.12214,48.38657,55.099594,51.36667]\n",
       "},\n",
       "\"mapping\":{\n",
       "\"x\":\"from_lon\",\n",
       "\"y\":\"from_lat\",\n",
       "\"xend\":\"to_lon\",\n",
       "\"yend\":\"to_lat\",\n",
       "\"color\":\"path\"\n",
       "},\n",
       "\"tooltips\":\"none\",\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"arrow\":{\n",
       "\"name\":\"arrow\",\n",
       "\"angle\":15.0,\n",
       "\"length\":8.0,\n",
       "\"type\":\"closed\"\n",
       "},\n",
       "\"curvature\":0.2,\n",
       "\"spacer\":2.5,\n",
       "\"size_start\":3.0,\n",
       "\"size_end\":3.0,\n",
       "\"size\":0.5\n",
       "},{\n",
       "\"geom\":\"point\",\n",
       "\"data\":{\n",
       "\"name\":[\"Erzurum\",\"Boldino\",\"Vitebsk\",\"Vladikavkaz\",\"Georgiyevsk\",\"Goryachie Vody\",\"Ekaterinodar\",\"Ekaterinoslav\",\"Kazan\",\"Kars\",\"Kiev\",\"Chișinău\",\"Moscow\",\"Mikhaylovskoye\",\"Nizhny Novgorod\",\"Novocherkassk\",\"Odessa\",\"Orenburg\",\"Oryol\",\"Petersburg\",\"Polotnyany Zavod\",\"Pskov\",\"Svyatogorskij monastery\",\"Simbirsk\",\"Simferopol\",\"Stavropol\",\"Tiflis\",\"Uralsk\",\"Chernihiv\"],\n",
       "\"latitude\":[39.90861,55.004477,55.187222,43.03667,44.1497667,44.04861,45.04484,48.450001,55.78874,40.60199,50.431782,47.00367,55.751244,57.061332,56.32753,47.414101,46.469391,51.776272,52.929697,59.93863,54.728958,57.8136,57.022162,54.32824,44.957813,45.0428,41.716667,51.23333,51.50551],\n",
       "\"longitude\":[41.27694,45.30885,30.205116,44.66778,43.4577689,43.05944,38.97603,34.983334,49.12214,43.09495,30.516382,28.907089,37.618423,28.919346,44.000717,40.110401,30.740883,55.099594,36.098689,30.31413,35.971598,28.3496,28.920945,48.38657,34.109547,41.9734,44.783333,51.36667,31.28487],\n",
       "\"years\":[\"1829\",\"1830, 1833\",\"1820, 1824\",\"1829\",\"1820, 1829\",\"1820\",\"1820\",\"1820\",\"1833\",\"1829\",\"1820-1821\",\"1820-1821, 1823\",\"1811, 1826-1830, 1833-1834, 1836\",\"1824, 1826\",\"1833\",\"1820, 1829\",\"1820-1821, 1823-1824\",\"1833\",\"1829\",\"1811, 1820, 1827-1828, 1833, 1836-1837\",\"1830, 1834\",\"1826\",\"1837\",\"1833\",\"1820\",\"1820, 1829\",\"1829\",\"1833\",\"1824\"]\n",
       "},\n",
       "\"mapping\":{\n",
       "\"x\":\"longitude\",\n",
       "\"y\":\"latitude\"\n",
       "},\n",
       "\"tooltips\":{\n",
       "\"formats\":[],\n",
       "\"lines\":[\"visited in|@years\"],\n",
       "\"title\":\"@name\"\n",
       "},\n",
       "\"data_meta\":{\n",
       "},\n",
       "\"size\":3.0,\n",
       "\"shape\":21.0,\n",
       "\"color\":\"black\",\n",
       "\"fill\":\"white\",\n",
       "\"alpha\":0.75\n",
       "}],\n",
       "\"metainfo_list\":[]\n",
       "};\n",
       "           var plotContainer = document.getElementById(\"DLbHX4\");\n",
       "           window.letsPlotCall(function() {{\n",
       "               LetsPlot.buildPlotFromProcessedSpecs(plotSpec, -1, -1, plotContainer);\n",
       "           }});\n",
       "       })();\n",
       "   </script>"
      ],
      "text/plain": [
       "<lets_plot.plot.core.PlotSpec at 0x7fb012c18f70>"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ggplot() \\\n",
    "    + geom_livemap(const_size_zoomin=3) \\\n",
    "    + geom_curve(aes(x='from_lon', y='from_lat', xend='to_lon', yend='to_lat', color='path'), data=df,\n",
    "                 size_start=3, size_end=3,\n",
    "                 spacer=2.5,\n",
    "                 size=0.5,\n",
    "                 curvature=0.2,\n",
    "                 arrow=arrow(type='closed', length=8, angle=15),\n",
    "                 tooltips='none') \\\n",
    "    + geom_point(aes(x='longitude', y='latitude'), data=places_df, \n",
    "                 size=3, shape=21, color='black', fill='white', alpha=0.75,\n",
    "                 tooltips=layer_tooltips().title('@name').line('visited in|@years')) \\\n",
    "    + scale_color_manual(name='trip name', values=['#addd8e', '#e34a33', '#8856a7', '#2c7fb8',\n",
    "                                                   '#1c9099', '#006d2c', '#fec44f', '#636363']) \\\n",
    "    + coord_cartesian(xlim=[26, 58], ylim=[38, 62]) \\\n",
    "    + ggtitle(\"Alexander Pushkin's Trips\") \\\n",
    "    + ggsize(1000, 800) \\\n",
    "    + theme_void() \\\n",
    "    + theme(legend_position=[1, 1], legend_justification=[1, 1], legend_title=\"blank\")"
   ]
  }
 ],
 "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.8.18"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}