Anotar Seta de Texto

Introdução

Este tutorial irá guiá-lo através da criação de um gráfico de dispersão anotado com uma seta de texto usando Matplotlib em Python.

Dicas para a VM (Máquina Virtual)

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 bibliotecas e gerar dados aleatórios

Primeiramente, precisamos importar as bibliotecas necessárias e gerar alguns dados aleatórios para o nosso gráfico de dispersão.

import matplotlib.pyplot as plt
import numpy as np

## Fixing random state for reproducibility
np.random.seed(19680801)

fig, ax = plt.subplots(figsize=(5, 5))
ax.set_aspect(1)

x1 = -1 + np.random.randn(100)
y1 = -1 + np.random.randn(100)
x2 = 1. + np.random.randn(100)
y2 = 1. + np.random.randn(100)

ax.scatter(x1, y1, color="r")
ax.scatter(x2, y2, color="g")

Adicionar anotações de texto ao gráfico

Em seguida, adicionaremos anotações de texto ao gráfico usando a função ax.text(). Criaremos duas anotações, uma para "Sample A" (Amostra A) e outra para "Sample B" (Amostra B).

bbox_props = dict(boxstyle="round", fc="w", ec="0.5", alpha=0.9)
ax.text(-2, -2, "Sample A", ha="center", va="center", size=20,
        bbox=bbox_props)
ax.text(2, 2, "Sample B", ha="center", va="center", size=20,
        bbox=bbox_props)

Adicionar uma seta de texto para indicar a direção

Para indicar a direção dos dados, adicionaremos uma seta de texto usando a função ax.text() e o parâmetro bbox com o boxstyle definido como "rarrow".

bbox_props = dict(boxstyle="rarrow", fc=(0.8, 0.9, 0.9), ec="b", lw=2)
t = ax.text(0, 0, "Direction", ha="center", va="center", rotation=45,
            size=15,
            bbox=bbox_props)

bb = t.get_bbox_patch()
bb.set_boxstyle("rarrow", pad=0.6)

Definir os limites do gráfico e mostrar o gráfico

Finalmente, definiremos os limites x e y do gráfico e mostraremos o gráfico usando as funções ax.set_xlim(), ax.set_ylim() e plt.show().

ax.set_xlim(-4, 4)
ax.set_ylim(-4, 4)

plt.show()

Resumo

Neste tutorial, aprendemos como criar um gráfico de dispersão anotado com uma seta de texto usando Matplotlib em Python. Usamos a função ax.text() para adicionar anotações e uma seta de texto ao gráfico, e as funções ax.set_xlim(), ax.set_ylim() e plt.show() para definir os limites do gráfico e mostrar o gráfico.