Adicionar Barra de Cores ao Gráfico Matplotlib

Introdução

Matplotlib é uma biblioteca Python utilizada para visualização de dados. Neste laboratório, aprenderemos como adicionar uma barra de cores (colorbar) a um gráfico em Matplotlib. Barras de cores são úteis para indicar a gama de valores que um mapa de cores (colormap) representa.

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. Usaremos o módulo pyplot do Matplotlib, que fornece uma interface para criar gráficos.

import matplotlib.pyplot as plt

Criar um gráfico

Em seguida, criaremos um gráfico usando a função imshow do Matplotlib. Esta função exibe uma imagem no gráfico. Também criaremos uma figura com dois subplots.

fig, (ax1, ax2) = plt.subplots(1, 2)
fig.subplots_adjust(wspace=0.5)

im1 = ax1.imshow([[1, 2], [3, 4]])

im2 = ax2.imshow([[1, 2], [3, 4]])

Adicionar uma barra de cores ao gráfico

Agora, adicionaremos uma barra de cores a cada subplot usando a função make_axes_locatable do Matplotlib. Esta função recebe um eixo existente, adiciona-o a um novo AxesDivider e retorna o AxesDivider. O método append_axes do AxesDivider pode então ser usado para criar um novo eixo em um determinado lado ("top", "right", "bottom" ou "left") do eixo original.

ax1_divider = make_axes_locatable(ax1)
cax1 = ax1_divider.append_axes("right", size="7%", pad="2%")
cb1 = fig.colorbar(im1, cax=cax1)

ax2_divider = make_axes_locatable(ax2)
cax2 = ax2_divider.append_axes("top", size="7%", pad="2%")
cb2 = fig.colorbar(im2, cax=cax2, orientation="horizontal")
cax2.xaxis.set_ticks_position("top")

Exibir o gráfico

Finalmente, exibiremos o gráfico usando a função show do Matplotlib.

plt.show()

Resumo

Neste laboratório, aprendemos como adicionar uma barra de cores a um gráfico no Matplotlib. Usamos a função make_axes_locatable para adicionar eixos adicionais ao gráfico e a função colorbar para criar a barra de cores. Também aprendemos como alterar a orientação e a posição da barra de cores.