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.