Ei, William! Como vai?
Parabéns por concluir o Desafio 2, posso ver atenção ao processo de visualização. O seu código está certo e muito organizado, isso ajuda demais na legibilidade.
Para complementar os estudos, segue um artigo sobre Matplotlib:
Continue se dedicando aos estudos e qualquer dúvida, conte conosco.