Introdução
Neste tutorial, aprenderemos como criar um gráfico de dispersão (scatter plot) com cores e tamanhos de marcadores variáveis usando Python Matplotlib.
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, que são Matplotlib e Numpy.
import matplotlib.pyplot as plt
import numpy as np
Carregar dados
Carregaremos um array de registro (record array) numpy a partir de dados CSV do Yahoo com os campos date, open, high, low, close, volume, adj_close do diretório mpl-data/sample_data. O array de registro armazena a data como um np.datetime64 com uma unidade de dia ('D') na coluna date.
import matplotlib.cbook as cbook
price_data = cbook.get_sample_data('goog.npz')['price_data'].view(np.recarray)
price_data = price_data[-250:] ## get the most recent 250 trading days
Calcular valores para o gráfico de dispersão (scatter plot)
Calcularemos os valores delta1, volume e close para o gráfico de dispersão.
delta1 = np.diff(price_data.adj_close) / price_data.adj_close[:-1]
## Marker size in units of points^2
volume = (15 * price_data.volume[:-2] / price_data.volume[0])**2
close = 0.003 * price_data.close[:-2] / 0.003 * price_data.open[:-2]
Criar gráfico de dispersão (scatter plot)
Criaremos um gráfico de dispersão com cores e tamanhos de marcadores variados usando os valores calculados.
fig, ax = plt.subplots()
ax.scatter(delta1[:-1], delta1[1:], c=close, s=volume, alpha=0.5)
ax.set_xlabel(r'$\Delta_i$', fontsize=15)
ax.set_ylabel(r'$\Delta_{i+1}$', fontsize=15)
ax.set_title('Volume and percent change')
ax.grid(True)
fig.tight_layout()
plt.show()
Resumo
Aprendemos como criar um gráfico de dispersão (scatter plot) com cores e tamanhos de marcadores variados usando Python Matplotlib. Começamos importando as bibliotecas necessárias, depois carregamos os dados, calculamos os valores para o gráfico de dispersão e criamos o gráfico de dispersão.