[Projeto] Estava me dando agonia repetir o mesmo plot 4 vezes, então criei uma função pra plotar o gráfico de uma vez | Data Visualization: criando gráficos com bibliotecas Python

[Projeto] Estava me dando agonia repetir o mesmo plot 4 vezes, então criei uma função pra plotar o gráfico de uma vez

top_4_paises = ['Brasil', 'Argentina', 'Colômbia', 'Peru']
fig, ax = plt.subplots(figsize=(10,5))

def gera_grafico_de_linha(df, pais):

  ax = sns.lineplot(df.loc[pais, anos], label=pais, lw=3)

def gera_grafico(df, list):

  for pais in list:
    gera_grafico_de_linha(df, pais)

  ax.set_title('Imigração dos maiores países da América do Sul\npara o Canadá de 1980 a 2013', loc='left', fontsize=20)
  ax.set_xlabel('Ano', fontsize=14)
  ax.set_ylabel('Número de imigrações', fontsize=14)

  ax.xaxis.set_major_locator(plt.MultipleLocator(5))
  ax.legend(title='Países', loc='upper right', bbox_to_anchor=(1.18, 1.02))

  plt.show()

gera_grafico(america_sul, top_4_paises)