Visualização de Gridspecs Aninhados com Matplotlib

Introdução

Matplotlib é uma biblioteca de visualização de dados em Python. Ela oferece uma ampla variedade de gráficos e diagramas que podem ser usados para representar dados de diversas formas. Neste laboratório, vamos percorrer o processo de criação de gridspecs aninhados usando 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 a Biblioteca Matplotlib

O primeiro passo é importar a biblioteca Matplotlib. Também usaremos o módulo gridspec do Matplotlib.

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

Criar a Figura e o GridSpec Externo

O próximo passo é criar uma figura e um gridspec externo. Neste exemplo, criaremos um gridspec de 1 por 2.

fig = plt.figure()
gs0 = gridspec.GridSpec(1, 2, figure=fig)

Criar o GridSpec Interno

Agora, criaremos o gridspec interno. Usaremos o método GridSpecFromSubplotSpec para criar um gridspec de 3 por 3 que será um subplot do gridspec externo.

gs00 = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=gs0[0])

Adicionar Subplots ao GridSpec Interno

Agora adicionaremos subplots ao gridspec interno. Criaremos três subplots: ax1, ax2 e ax3.

ax1 = fig.add_subplot(gs00[:-1, :])
ax2 = fig.add_subplot(gs00[-1, :-1])
ax3 = fig.add_subplot(gs00[-1, -1])

Criar Outro GridSpec Interno

Agora criaremos outro gridspec interno. Desta vez, usaremos o método subgridspec para criar um gridspec de 3 por 3 que será um subplot da segunda coluna do gridspec externo.

gs01 = gs0[1].subgridspec(3, 3)

Adicionar Subplots ao Segundo GridSpec Interno

Agora adicionaremos subplots ao segundo gridspec interno. Criaremos três subplots: ax4, ax5 e ax6.

ax4 = fig.add_subplot(gs01[:, :-1])
ax5 = fig.add_subplot(gs01[:-1, -1])
ax6 = fig.add_subplot(gs01[-1, -1])

Formatar os Eixos

Formataremos os eixos de todos os subplots usando a função format_axes. Esta função adicionará um rótulo de texto a cada subplot e removerá os rótulos de marcação (tick labels).

def format_axes(fig):
    for i, ax in enumerate(fig.axes):
        ax.text(0.5, 0.5, "ax%d" % (i+1), va="center", ha="center")
        ax.tick_params(labelbottom=False, labelleft=False)

format_axes(fig)

Exibir a Figura

Finalmente, exibiremos a figura usando o método show.

plt.show()

Resumo

Neste laboratório, aprendemos como criar gridspecs aninhados usando Matplotlib. Criamos um gridspec externo e dois gridspecs internos para criar um layout complexo de subplots. Também aprendemos como formatar os eixos dos subplots usando uma função personalizada.