Introdução
Neste tutorial, aprenderemos como usar a biblioteca matplotlib para criar barras de cores (colorbars) para visualizações. As barras de cores são uma ferramenta útil para ajudar a interpretar visualizações, fornecendo uma escala de cores que corresponde aos dados que estão sendo plotados. Usaremos matplotlib para criar barras de cores para visualizações com valores de dados positivos e negativos.
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çamos importando as bibliotecas necessárias: numpy e matplotlib.pyplot.
import matplotlib.pyplot as plt
import numpy as np
Gerar Dados
Geramos alguns dados de amostra para plotar, usando a função mgrid do numpy.
## setup some generic data
N = 37
x, y = np.mgrid[:N, :N]
Z = (np.cos(x*0.2) + np.sin(y*0.3))
Criar um Gráfico de Dados Positivos e Colorbar
Criamos um gráfico dos dados positivos e adicionamos uma colorbar ao gráfico usando a função colorbar.
## plot just the positive data and save the
## color "mappable" object returned by ax1.imshow
pos = plt.imshow(Zpos, cmap='Blues', interpolation='none')
## add the colorbar using the figure's method,
## telling which mappable we're talking about and
## which axes object it should be near
plt.colorbar(pos)
Criar um Gráfico de Dados Negativos e Colorbar
Criamos um gráfico dos dados negativos e adicionamos uma colorbar ao gráfico usando a função colorbar. Desta vez, especificamos a localização da colorbar, bem como os parâmetros anchor e shrink.
## repeat everything above for the negative data
## you can specify location, anchor and shrink the colorbar
neg = plt.imshow(Zneg, cmap='Reds_r', interpolation='none')
plt.colorbar(neg, location='right', anchor=(0, 0.3), shrink=0.7)
Criar um Gráfico com Dados Positivos e Negativos
Criamos um gráfico com dados positivos e negativos e adicionamos uma colorbar ao gráfico usando a função colorbar. Desta vez, especificamos os valores mínimo e máximo para a colorbar usando os parâmetros vmin e vmax.
## Plot both positive and negative values between +/- 1.2
pos_neg_clipped = plt.imshow(Z, cmap='RdBu', vmin=-1.2, vmax=1.2,
interpolation='none')
## Add minorticks on the colorbar to make it easy to read the
## values off the colorbar.
cbar = plt.colorbar(pos_neg_clipped, extend='both')
cbar.minorticks_on()
Resumo
Neste tutorial, aprendemos como usar a biblioteca matplotlib para criar colorbars para visualizações. Cobrimos como criar colorbars para visualizações com valores de dados positivos e negativos. Com estas ferramentas, podemos criar visualizações mais informativas e úteis.