Funzione definita dall’utente per creare i grafici

Nell'ultimo esercizio, hai scritto questo codice per creare il grafico:

# Sottocampiona aziende tech e fmcg
subset_dat = dataset.loc[dataset["comp_type"].isin(["tech", "fmcg"])]

# Calcola la media annuale del margine lordo per aziende tech e fmcg
subset_dat_avg = subset_dat.pivot_table(index=["Year", "comp_type"], values = "gross_margin").reset_index()

# Aggiungi la colonna company
subset_dat_avg["company"] = np.where(subset_dat_avg["comp_type"]=="tech", "Avg tech", "Avg fmcg")

# Concatena i DataFrame
plot_df = pd.concat([subset_dat, subset_dat_avg], axis=0)

# Crea il grafico
sns.relplot(data=plot_df.reset_index(drop=True), x="Year", y="gross_margin", hue="company", col="comp_type", kind="line")
plt.show()
plt.close()

Nota che in questo esercizio eseguiamo le stesse azioni sui DataFrame di tech e FMCG. È ripetitivo e va contro un principio di coding chiamato DRY - Don't repeat yourself (non ripeterti). Il codice ripetitivo è sconsigliato perché aumenta il lavoro e rende il codice più soggetto a errori. In questo esercizio, definirai una tua funzione per elaborare i dati e creare i grafici.