Introdução
Matplotlib é uma poderosa biblioteca de visualização de dados em Python. Neste laboratório, exploraremos o uso de EllipseCollection para desenhar uma coleção de elipses.
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
Começaremos importando as bibliotecas necessárias.
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.collections import EllipseCollection
Criar dados para as elipses
Criamos dados para nossas elipses na forma de arrays de coordenadas x, coordenadas y, largura, altura e ângulo.
x = np.arange(10)
y = np.arange(15)
X, Y = np.meshgrid(x, y)
XY = np.column_stack((X.ravel(), Y.ravel()))
ww = X / 10.0
hh = Y / 15.0
aa = X * 9
Criar a Coleção de Elipses
Criamos um EllipseCollection com os dados acima e especificamos as unidades como 'x' e os deslocamentos como XY.
fig, ax = plt.subplots()
ec = EllipseCollection(ww, hh, aa, units='x', offsets=XY,
offset_transform=ax.transData)
Definir a cor das elipses
Definimos a cor de cada elipse no EllipseCollection com base na soma de suas coordenadas x e y.
ec.set_array((X + Y).ravel())
Adicionar a coleção ao gráfico
Adicionamos o EllipseCollection ao gráfico.
ax.add_collection(ec)
ax.autoscale_view()
ax.set_xlabel('X')
ax.set_ylabel('y')
cbar = plt.colorbar(ec)
cbar.set_label('X+Y')
plt.show()
Resumo
Neste laboratório, aprendemos como usar EllipseCollection para desenhar uma coleção de elipses em Matplotlib. Também aprendemos como definir a cor de cada elipse com base em suas coordenadas x e y.