Anotações Simples em Matplotlib

Introdução

Este laboratório irá guiá-lo através do processo de adição de anotações aos seus gráficos Matplotlib. As anotações ajudam a destacar pontos de dados específicos ou fornecer informações adicionais ao visualizador. As anotações podem incluir texto, setas e formas. Você aprenderá como adicionar anotações ao seu gráfico, personalizá-las e posicioná-las.

Dicas da VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook para 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 Matplotlib

Antes de começarmos a trabalhar com Matplotlib, precisamos importá-lo. O código a seguir importará Matplotlib e nos permitirá usar suas funções de plotagem.

import matplotlib.pyplot as plt

Criar um Gráfico

Agora criaremos um gráfico para anotar. O código a seguir criará um gráfico com dois pontos de dados.

fig, ax = plt.subplots()
x = [1, 2]
y = [3, 4]
ax.plot(x, y, "o")

Adicionar Anotação de Texto

Agora adicionaremos uma anotação de texto ao gráfico. O código a seguir adicionará o texto "Data Point 1" no primeiro ponto de dados.

ax.annotate("Data Point 1", xy=(1, 3), xytext=(1.5, 3.5),
            arrowprops=dict(facecolor="black", shrink=0.05))

Adicionar Anotação de Seta

Agora adicionaremos uma anotação de seta ao gráfico. O código a seguir adicionará uma seta do primeiro ponto de dados para o segundo ponto de dados.

ax.annotate("", xy=(1, 3), xytext=(2, 4),
            arrowprops=dict(arrowstyle="->", connectionstyle="arc3"))

Adicionar Anotação de Forma

Agora adicionaremos uma anotação de forma ao gráfico. O código a seguir adicionará um retângulo em torno do segundo ponto de dados.

bbox = dict(boxstyle="round", fc="0.8")
ax.annotate("Data Point 2", xy=(2, 4), xytext=(2.5, 4.5),
            bbox=bbox,
            arrowprops=dict(facecolor="black", shrink=0.05))

Personalizar Anotações

Podemos personalizar as anotações alterando o tamanho da fonte, a cor da fonte e o estilo da seta. O código a seguir alterará o tamanho da fonte, a cor da fonte e o estilo da seta da anotação de texto.

ax.annotate("Data Point 1", xy=(1, 3), xytext=(1.5, 3.5),
            arrowprops=dict(facecolor="black", shrink=0.05, arrowstyle="->"),
            fontsize=12, color="red")

Posicionar Anotações

Podemos posicionar as anotações usando diferentes sistemas de coordenadas. O código a seguir posicionará a anotação de texto usando coordenadas de dados e a anotação de seta usando coordenadas da figura.

ax.annotate("Data Point 1", xy=(1, 3), xytext=(1.5, 3.5),
            arrowprops=dict(facecolor="black", shrink=0.05),
            xycoords="data", textcoords="data")
ax.annotate("", xy=(1, 3), xytext=(0.5, 0.5),
            arrowprops=dict(facecolor="black", shrink=0.05),
            xycoords="data", textcoords="figure fraction")

Resumo

Neste laboratório, você aprendeu como adicionar anotações aos seus gráficos Matplotlib. Você aprendeu como adicionar anotações de texto, seta e forma, personalizá-las e posicioná-las. As anotações ajudam a destacar pontos de dados específicos ou fornecer informações adicionais ao visualizador.