Gráficos de Dispersão

Parte da seção Dados e gráficos do Journey de Machine Learning da Coddy. Lição 21 de 60.

Dez plantas de interior: horas de sol por dia e quanto cada uma cresceu. Ligadas por uma linha, elas formam um emaranhado. Como pontos, surge um padrão:

Dois gráficos das mesmas dez plantas, com o sol no eixo horizontal e o crescimento no vertical. À esquerda, plt.plot as conecta em um emaranhado em zigue-zague. À direita, plt.scatter mostra pontos que sobem para a direita

plt.scatter(x, y) desenha um ponto para cada par, sem linhas:

sun = [2, 6, 3, 8, 5, 1, 7, 4, 6, 3]
growth = [4, 10, 7, 12, 6, 2, 9, 8, 7, 3]
plt.scatter(sun, growth)
plt.xlabel("Sun (hours a day)")
plt.ylabel("Growth (cm)")
plt.show()

Mais sol, mais crescimento. Um gráfico de dispersão mostra como duas medidas de amostras separadas se relacionam: a primeira análise antes que um modelo preveja uma a partir da outra.

Valores em ordem, como os dias da semana, ainda usam plt.plot.

Desafio

Médio

Um pescador registrou cada peixe na ordem em que foi pescado: uma linha por peixe, comprimento (cm) e depois peso (g). O código inicial os conecta com plt.plot, então a linha ziguezagueia e oculta o padrão.

Desenhe cada peixe como um ponto: comprimento no eixo horizontal, peso no eixo vertical.
Não deixe nenhuma linha.
Mantenha o título, os rótulos e plt.show().

O seu deve ficar parecido com este, de outra pescaria:

Um gráfico de dispersão intitulado Fish Catch: nove pontos subindo de 18 cm e 90 g até 42 cm e 980 g, sem linha

Seu programa não imprime nada. Coddy descreve seu gráfico na saída.

Experimente você mesmo

import numpy as np
import matplotlib.pyplot as plt

# Cada linha representa um peixe, na ordem em que foi capturado: comprimento (cm), peso (g)
fish = np.array([[22, 180], [30, 390], [25, 250], [35, 620],
                 [28, 330], [40, 870], [19, 120], [33, 510]])

# Corrija o gráfico abaixo
plt.plot(fish[:, 0], fish[:, 1])

# Não altere as linhas abaixo
plt.title("Fish Catch")
plt.xlabel("Length (cm)")
plt.ylabel("Weight (g)")
plt.show()