Add legend to the map

1 files changed, 5 insertions, 0 deletions

diff --git a/countries_motives_sectors_tables.py b/countries_motives_sectors_tables.py
index f5d8e51..2ba2fc6 100755
--- a/countries_motives_sectors_tables.py
+++ b/countries_motives_sectors_tables.py
@@ -94,6 +94,7 @@ def plot_map(countries, targeting_percentages, dst_path="dst.svg"):
     import geopandas as gpd
     import matplotlib.pyplot as plt
     from matplotlib.colors import ListedColormap
+    from matplotlib.patches import Patch
     import pandas as pd
     import math
     from pathlib import Path
@@ -102,6 +103,10 @@ def plot_map(countries, targeting_percentages, dst_path="dst.svg"):
     world = gpd.read_file(gpd.datasets.get_path('naturalearth_lowres'))
     world.plot(column='name', cmap=ListedColormap([[0.7, 0.7, 0.7]]))
 
+    handles = [Patch(color=color, label=origin_label)
+               for origin_label, color
+               in zip(sorted(origin_labels.values()), chart_colors)]
+    plt.legend(handles=handles, loc="lower right", fontsize="xx-small")
     plt.axis("off")
     plt.margins(x=0, y=0)
     plt.savefig("empty_world_map.svg", bbox_inches='tight', pad_inches=0)