Adicionar Barra de Cores a Eixos Inseridos | Tutorial Matplotlib Python

Introdução

Neste laboratório, você aprenderá como adicionar uma barra de cores (colorbar) a eixos inseridos usando Matplotlib em Python. Uma barra de cores é uma representação visual do mapeamento de uma gama de cores para uma gama de valores numéricos. Um eixo inserido (inset axis) é um eixo menor que é colocado dentro do eixo maior de um gráfico.

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 Bibliotecas e Dados

Primeiramente, importe as bibliotecas e os dados necessários que serão usados no gráfico.

import matplotlib.pyplot as plt
from matplotlib import cbook
from mpl_toolkits.axes_grid1.inset_locator import inset_axes, zoomed_inset_axes

fig, ax = plt.subplots(figsize=[5, 4])

Z = cbook.get_sample_data("axes_grid/bivariate_normal.npy")
extent = (-3, 4, -4, 3)

Definir o Gráfico Principal

Defina o gráfico principal ajustando a proporção (aspect ratio) e os limites dos eixos x e y.

ax.set(aspect=1, xlim=(-15, 15), ylim=(-20, 5))

Criar um Eixo Inserido (Inset Axis)

Crie um eixo inserido usando a função zoomed_inset_axes. Defina o nível de zoom e a localização do eixo inserido dentro do gráfico principal.

axins = zoomed_inset_axes(ax, zoom=2, loc='upper left')
axins.set(xticks=[], yticks=[])

Adicionar uma Imagem ao Eixo Inserido (Inset Axis)

Adicione uma imagem ao eixo inserido usando a função imshow. Defina a extensão (extent) e a origem da imagem.

im = axins.imshow(Z, extent=extent, origin="lower")

Adicionar uma Barra de Cores (Colorbar)

Adicione uma barra de cores ao eixo inserido usando a função inset_axes. Defina a largura, altura, localização e bounding box da barra de cores.

cax = inset_axes(axins,
                 width="5%",  ## width = 10% of parent_bbox width
                 height="100%",  ## height : 50%
                 loc='lower left',
                 bbox_to_anchor=(1.05, 0., 1, 1),
                 bbox_transform=axins.transAxes,
                 borderpad=0,
                 )
fig.colorbar(im, cax=cax)

Exibir o Gráfico (Plot)

Exiba o gráfico usando a função show.

plt.show()

Resumo

Parabéns! Você aprendeu com sucesso como adicionar uma barra de cores (colorbar) a eixos inseridos usando Matplotlib em Python. Esta é uma técnica útil para visualizar dados de uma forma mais detalhada e informativa. Lembre-se de ajustar os parâmetros de acordo com suas necessidades e preferências específicas.

Adicionando uma Barra de Cores a Eixos Inseridos