Oii, como você está?
Seu código está bem estruturado, parabéns!
Uma melhoria poderia ser feita, por exemplo, na criação do dataframe de comparação. Veja essa alternativa:
df_comparacao = df.loc[['Brasil', 'Argentina'], anos]
Assim temos um dataframe criado mais rapidamente. Para transpor, ou seja, obter cada valor como colunas, é possível utilizar a propriedade T, assim:
df_comparacao = df_comparacao.T
O próximo passo já é plotar o gráfico e fazer as configurações de estilo!
Bons estudos e um grande abraço!
Conte conosco.
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!