Criando Subplots com Matplotlib

Introdução

Matplotlib é uma popular biblioteca de visualização de dados em Python. Ela oferece uma variedade de funções para criar diferentes tipos de gráficos. Uma de suas principais características é a capacidade de criar subplots. Isso permite que os usuários criem múltiplos gráficos na mesma figura, facilitando a comparação de diferentes conjuntos de dados ou visualizações dos mesmos dados. Neste laboratório, vamos percorrer o processo de criação de subplots 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ê.

Criando uma Figura com um Único Subplot

A maneira mais simples de criar um único subplot é usando a função subplots() sem nenhum argumento. Esta função retorna um objeto Figure e um único objeto Axes.

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

Empilhando Subplots em Uma Direção

Para criar múltiplos subplots empilhados verticalmente ou horizontalmente, podemos passar o número de linhas e colunas como argumentos para a função subplots(). O objeto axs retornado é um array numpy 1D contendo a lista de Axes criados.

fig, axs = plt.subplots(2)
axs[0].plot(x, y)
axs[1].plot(x, -y)

Empilhando Subplots em Duas Direções

Para criar uma grade de subplots, podemos passar o número de linhas e colunas como argumentos para a função subplots(). O objeto axs retornado é um array NumPy 2D.

fig, axs = plt.subplots(2, 2)
axs[0, 0].plot(x, y)
axs[0, 1].plot(x, y, 'tab:orange')
axs[1, 0].plot(x, -y, 'tab:green')
axs[1, 1].plot(x, -y, 'tab:red')

Compartilhando Eixos

Por padrão, cada Axes é escalado individualmente. Para alinhar o eixo horizontal ou vertical de subplots, podemos usar os parâmetros sharex ou sharey.

fig, (ax1, ax2) = plt.subplots(2, sharex=True)
ax1.plot(x, y)
ax2.plot(x + 1, -y)

Eixos Polares

Podemos criar uma grade de Axes polares passando o parâmetro projection='polar' para a função subplots().

fig, (ax1, ax2) = plt.subplots(1, 2, subplot_kw=dict(projection='polar'))
ax1.plot(x, y)
ax2.plot(x, y ** 2)

Resumo

Neste laboratório, aprendemos como criar subplots usando Matplotlib. Cobrimos a criação de uma figura com um único subplot, o empilhamento de subplots em uma direção, o empilhamento de subplots em duas direções, o compartilhamento de eixos e a criação de eixos polares. Ao usar essas técnicas, podemos criar visualizações complexas que nos permitem comparar e analisar múltiplos conjuntos de dados de uma só vez.