Criando Subplots Matplotlib em Python

Introdução

Matplotlib é uma biblioteca Python utilizada para criar visualizações como gráficos de linhas, gráficos de dispersão (scatter plots), gráficos de barras e subplots, entre outros. Neste laboratório, você aprenderá como criar uma figura com dois subplots usando .pyplot.subplot.

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

Para este tutorial, usaremos o módulo pyplot da biblioteca Matplotlib e a biblioteca numpy.

import matplotlib.pyplot as plt
import numpy as np

Definir os dados

Definiremos dois conjuntos de dados que usaremos para criar nossos subplots.

def f(t):
    return np.exp(-t) * np.cos(2*np.pi*t)

t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)

Criar os subplots

Criaremos uma figura com dois subplots usando .pyplot.subplot.

plt.figure()

plt.subplot(211)
plt.plot(t1, f(t1), color='tab:blue', marker='o')
plt.plot(t2, f(t2), color='black')

plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), color='tab:orange', linestyle='--')

plt.show()

A função subplot() recebe três argumentos: o número de linhas, o número de colunas e o índice do gráfico atual. O índice começa em 1 no canto superior esquerdo e aumenta linha por linha. Neste exemplo, criamos uma figura com dois subplots: um na parte superior e outro na parte inferior.

No primeiro subplot, plotamos t1 contra f(t1) e t2 contra f(t2). Definimos a cor do primeiro gráfico como azul e adicionamos marcadores circulares a cada ponto de dados. Definimos a cor do segundo gráfico como preto.

No segundo subplot, plotamos t2 contra a função cosseno de 2*np.pi*t2. Definimos a cor do gráfico como laranja e o estilo da linha como tracejado.

Referências

O uso das seguintes funções, métodos, classes e módulos é demonstrado neste exemplo:

• matplotlib.pyplot.figure
• matplotlib.pyplot.subplot

Resumo

Neste laboratório, aprendemos como criar uma figura com dois subplots usando .pyplot.subplot. Definimos dois conjuntos de dados e os plotamos em dois subplots usando diferentes cores e estilos de linha. Também fornecemos referências às funções e módulos usados neste exemplo.