Criar Legendas em Plots Matplotlib

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.