{
"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",
" \n",
" \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": [
" \n",
" "
],
"text/plain": [
""
]
},
"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
}