Introdução
Na visualização de dados, uma legenda (legend) é a chave para interpretar os elementos visuais de um gráfico. Ela ajuda o observador a entender os dados e o significado da representação visual. Matplotlib é uma biblioteca Python popular para criar visualizações de dados, incluindo gráficos com legendas. Neste tutorial, aprenderemos como criar uma legenda com rótulos (labels) predefinidos em Matplotlib.
Dicas para a VM
Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook e acessar o Jupyter Notebook para praticar.
Às vezes, pode ser necessário aguardar alguns segundos para que o Jupyter Notebook termine de carregar. A validação das operações não pode ser automatizada devido a limitações no Jupyter Notebook.
Se você enfrentar problemas durante o aprendizado, sinta-se à vontade para perguntar ao Labby. Forneça feedback após a sessão, e resolveremos o problema prontamente para você.
Importando as Bibliotecas Necessárias
Começaremos importando as bibliotecas necessárias, que incluem Matplotlib e NumPy. Usamos NumPy para gerar alguns dados falsos para nosso gráfico.
import matplotlib.pyplot as plt
import numpy as np
Gerando os Dados
Em seguida, geraremos alguns dados falsos para usar em nosso gráfico. Criaremos dois arrays, a e b, usando a função arange do NumPy. Em seguida, calculamos mais dois arrays, c e d, usando a função exp para computar a exponencial de a e d como o inverso de c.
## Make some fake data.
a = b = np.arange(0, 3, .02)
c = np.exp(a)
d = c[::-1]
Criando o Gráfico
Agora estamos prontos para criar nosso gráfico. Usaremos a função plot do Matplotlib para plotar três linhas no mesmo gráfico, cada uma com um rótulo predefinido. Usaremos o parâmetro label para atribuir os rótulos a cada linha.
## Create plots with pre-defined labels.
fig, ax = plt.subplots()
ax.plot(a, c, 'k--', label='Model length')
ax.plot(a, d, 'k:', label='Data length')
ax.plot(a, c + d, 'k', label='Total message length')
Adicionando a Legenda
Para adicionar a legenda ao nosso gráfico, usamos a função legend do Matplotlib. Passamos o parâmetro loc para especificar a localização da legenda e o parâmetro shadow para adicionar um efeito de sombra à legenda. Também usamos o parâmetro fontsize para definir o tamanho da fonte da legenda.
legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')
Estilizando a Legenda
Finalmente, podemos estilizar a legenda para torná-la mais visualmente atraente. Usamos a função get_frame para obter o frame da legenda e, em seguida, usamos a função set_facecolor para definir a cor de fundo do frame.
## Put a nicer background color on the legend.
legend.get_frame().set_facecolor('C0')
Exibindo o Gráfico
Agora podemos exibir o gráfico usando a função show do Matplotlib.
plt.show()
Resumo
Neste tutorial, aprendemos como criar uma legenda com rótulos predefinidos no Matplotlib. Usamos a função plot para plotar três linhas no mesmo gráfico e usamos o parâmetro label para atribuir rótulos a cada linha. Em seguida, usamos a função legend para adicionar a legenda ao gráfico e estilizamos a legenda para torná-la mais visualmente atraente. Seguindo estes passos, você pode criar legendas para seus próprios gráficos no Matplotlib.