#tutorial 
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D
import pandas as pd


color_dict = {"Norway": "#2B314D", "Denmark": "#A54836", "Sweden": "#5375D4"}

xy_ticklabel_color, title_color, grid_color, datalabels_color ='#101628',"#101628", "#C8C9C9", "#101628"

data = {
    "year": [2004, 2022, 2004, 2022, 2004, 2022],
    "countries" : [ "Denmark", "Denmark", "Norway", "Norway","Sweden", "Sweden"],
    "sites": [4,10,5,8,13,15]
}
df= pd.DataFrame(data)
df['sub_total'] = df.groupby('year')['sites'].transform('sum')
df = df.sort_values(['countries' ,'year' ], ascending=True ).reset_index(drop=True)
#map the colors of a dict to a dataframe
df['color']= df.countries.map(color_dict)
df

fig, axes = plt.subplots(ncols = df.year.nunique(), nrows = 1,figsize=(5,5), facecolor = "#FFFFFF")

for (year, group),title, ax in zip(df.groupby("year"), titles, axes.ravel()):

    patches, texts, autotexts  =  ax.pie(
        group.sites, 
        labels = group.sites,
        autopct='%1.0f%%',    
        startangle=90,
        colors= group.color)

    ax.set_xlabel(
        group.sub_total.iloc[0],  
        color = title_color, 
        size = 14, 
        weight= "bold",
        labelpad = 23
        )

    ax.set_title(
        title, 
        size=10, 
        x=0.5, 
        y=1.2, 
        color= title_color
        )

    

    [autotext.set_color('white') for autotext in autotexts]

#add legend
lines = [Line2D([0], [0], color=c,  marker='o',linestyle='', markersize=10,) for c in df.color.unique()]

fig.legend(
    lines, 
    df.countries.unique().tolist(), 
    labelcolor= title_color,
    bbox_to_anchor= (0.5, -0), 
    loc = "lower center",
    ncols = 3,
    frameon = False, 
    fontsize = 8
    )

fig