Curso matplotlib - Várias séries

Em muitos casos vamos ter dois ou mais conjuntos de dados para serem plotados no gráfico, ou seja, várias séries de x e y (ou várias séries de y para o mesmo conjunto de x). No exemplo que esta sendo utilizado, com dados relacionando peso e altura dos cachorros, temos várias raças, e seria importante diferenciar as raças no gráfico.

Para adicionar mais de uma série em um gráfico de dispersão, basta adicionar um novo elemento de plt.scatter() com a nova série.

Por exemplo, o gráfico de dispersão que relaciona peso e altura apenas para a raça 'Chihuahua' (posição 0) é:

plt.figure(figsize=(8,6))
plt.scatter(x[0], y[0], label=raca_cachorro[0])
plt.legend()
plt.show()

Figura 1 - Gráfico de dispersão para a raça Chihuahua.

Já para a raça 'Poodle Toy' (posição 1):

plt.figure(figsize=(8,6))
plt.scatter(x[1], y[1], label=raca_cachorro[1])
plt.legend()
plt.show()

Figura 2 - Gráfico de dispersão para a raça Poodle Toy.

Para juntar estes dois gráficos (como séries diferentes), combinamos dois elementos de plt.scatter(), um após o outro:

plt.figure(figsize=(8,6))
plt.scatter(x[0], y[0], label=raca_cachorro[0])
plt.scatter(x[1], y[1], label=raca_cachorro[1])
plt.legend()
plt.show()

Figura 3 - Gráfico de dispersão para a raça Chihuahua e Poodle Toy.

Observe que o matplotlib irá automaticamente diferenciar a cor do marcador das diferentes séries de dados. É possível adicionar quantas séries quanto desejarmos:

plt.figure(figsize=(8,6))
plt.scatter(x[0], y[0], label=raca_cachorro[0])
plt.scatter(x[1], y[1], label=raca_cachorro[1])
plt.scatter(x[2], y[2], label=raca_cachorro[2])
plt.scatter(x[3], y[3], label=raca_cachorro[3])
plt.scatter(x[4], y[4], label=raca_cachorro[4])
plt.scatter(x[5], y[5], label=raca_cachorro[5])
plt.scatter(x[6], y[6], label=raca_cachorro[6])
plt.scatter(x[7], y[7], label=raca_cachorro[7])
plt.scatter(x[8], y[8], label=raca_cachorro[8])
plt.scatter(x[9], y[9], label=raca_cachorro[9])
plt.scatter(x[10], y[10], label=raca_cachorro[10])
plt.legend()
plt.show()

Figura 4 - Gráfico de dispersão para todas as raças de cachorro avaliadas.

Como adicionar diversos elementos de dispersão no matplotlib?

Adicionando vários elementos de plt.scatter() antes de apresentar o gráfico

Criando várias sequências de criação de gráficos (criar canvas, adicionar elemento, apresentar o gráfico)

Passando várias séries de dados através do parâmetro y em plt.scatter().

Desafio 1

Crie o mesmo gráfico da Figura 4, mas utilizando um loop for para adicionar as raças de cachorro no gráfico. São necessárias apenas 5 linhas de código.

Você pode acessar o notebook básico clicando aqui (mas tente criar um novo notebook do zero, e implementar todos os dados por conta própria).

O notebook com a resolução do desafio pode ser acessado clicando aqui. O vídeo da resolução do desafio 1 pode ser acessado neste link.

Anterior Próximo