Introdução
Matplotlib é uma biblioteca Python que permite aos usuários criar vários tipos de gráficos e plots. Este tutorial irá guiá-lo através do processo de criação de uma legenda em um plot 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ê.
Importar as bibliotecas necessárias
Antes de começarmos, precisamos importar as bibliotecas necessárias. Neste caso, usaremos a biblioteca Matplotlib.
import matplotlib.pyplot as plt
Criar uma figura e um subplot
Precisamos criar uma figura e um subplot para plotar nossos dados. Criaremos um plot com dois subplots.
fig = plt.figure()
ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")
ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")
Adicionar uma legenda ao plot
Agora adicionaremos uma legenda ao plot. Existem duas maneiras de adicionar uma legenda no Matplotlib. Usaremos ambos os métodos neste exemplo.
## Method 1: Place a legend above the subplot
ax.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc='lower left',
ncols=2, mode="expand", borderaxespad=0.)
## Method 2: Place a legend to the right of the subplot
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)
Exibir o plot
Finalmente, exibiremos o plot.
plt.show()
Resumo
Neste tutorial, aprendemos como adicionar uma legenda a um plot do Matplotlib. Usamos dois métodos diferentes para adicionar uma legenda. O primeiro método colocou a legenda acima do subplot, enquanto o segundo método colocou a legenda à direita do subplot.