Personalização de Subplots com GridSpec

Introdução

Neste tutorial, aprenderemos como usar GridSpec para gerar subplots, controlar os tamanhos relativos dos subplots com width_ratios e height_ratios, e o espaçamento ao redor e entre os subplots usando os parâmetros de subplot (left, right, bottom, top, wspace e hspace).

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

Começamos importando as bibliotecas necessárias, que são matplotlib.pyplot e GridSpec.

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

Gerar Subplots com GridSpec

Nesta etapa, usaremos GridSpec para gerar subplots. Criaremos uma figura com 2 linhas e 2 colunas. Também especificaremos os width_ratios e height_ratios para controlar os tamanhos relativos dos subplots.

fig = plt.figure()
gs = GridSpec(2, 2, width_ratios=[1, 2], height_ratios=[4, 1])
ax1 = fig.add_subplot(gs[0])
ax2 = fig.add_subplot(gs[1])
ax3 = fig.add_subplot(gs[2])
ax4 = fig.add_subplot(gs[3])

Controlar o Espaçamento ao Redor e Entre os Subplots

Nesta etapa, usaremos GridSpec para controlar o espaçamento ao redor e entre os subplots. Criaremos uma figura com 2 gridspecs, cada um com 3 linhas e 3 colunas. Especificaremos os parâmetros left, right, bottom, top, wspace e hspace para controlar o espaçamento.

fig = plt.figure()
gs1 = GridSpec(3, 3, left=0.05, right=0.48, wspace=0.05)
ax1 = fig.add_subplot(gs1[:-1, :])
ax2 = fig.add_subplot(gs1[-1, :-1])
ax3 = fig.add_subplot(gs1[-1, -1])

gs2 = GridSpec(3, 3, left=0.55, right=0.98, hspace=0.05)
ax4 = fig.add_subplot(gs2[:, :-1])
ax5 = fig.add_subplot(gs2[:-1, -1])
ax6 = fig.add_subplot(gs2[-1, -1])

Anotar Eixos

Nesta etapa, anotaremos os eixos com seus respectivos números de subplot.

def annotate_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)

annotate_axes(fig)

Exibir os Gráficos

Nesta etapa, exibiremos os gráficos.

plt.show()

Resumo

Neste tutorial, aprendemos como usar GridSpec para gerar subplots e controlar o espaçamento ao redor e entre os subplots. Também aprendemos como anotar os eixos com seus respectivos números de subplot. Essas habilidades serão úteis na criação de gráficos e visualizações complexas.