Tutorial de Geração de Subplots com Matplotlib

Introdução

Neste tutorial, você aprenderá como usar a função subplot2grid na biblioteca Matplotlib do Python para gerar subplots.

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

Antes de começarmos, precisamos importar a biblioteca Matplotlib usando o seguinte código:

import matplotlib.pyplot as plt

Criar um Objeto Figura

Para criar um objeto figura, use o seguinte código:

fig = plt.figure()

Definir Subplots Usando subplot2grid

Para definir subplots usando subplot2grid, primeiro precisamos especificar o tamanho da grade usando uma tupla com o número desejado de linhas e colunas. Também precisamos especificar a localização do subplot dentro da grade usando outra tupla.

Por exemplo, para criar uma grade 3x3 com um subplot que abrange toda a primeira linha e todas as três colunas, usamos o seguinte código:

ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=3)

Para criar um subplot que abrange a segunda linha e as duas primeiras colunas, usamos:

ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2)

Para criar um subplot que abrange as duas últimas linhas e a última coluna, usamos:

ax3 = plt.subplot2grid((3, 3), (1, 2), rowspan=2)

Para criar um subplot na última linha e na primeira coluna, usamos:

ax4 = plt.subplot2grid((3, 3), (2, 0))

Para criar um subplot na última linha e na segunda coluna, usamos:

ax5 = plt.subplot2grid((3, 3), (2, 1))

Anotar Eixos

Para anotar os eixos, podemos iterar pelos eixos da figura e adicionar texto usando a função text e a função tick_params para remover os rótulos das marcas de escala (tick labels).

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)

Exibir o Gráfico

Para exibir o gráfico, use o seguinte código:

plt.show()

Resumo

Neste tutorial, você aprendeu como usar a função subplot2grid em Matplotlib para gerar subplots. Você também aprendeu como criar um objeto figura (figure object), definir subplots dentro de uma grade, anotar os eixos e exibir o gráfico.