Adicionar Subplot a uma figura em Matplotlib

1. Adicionar Subplot a uma figura utilizando o método matplotlib.pyplot.figure.add_subplot()
2. Criar figura com subparcelas utilizando Matplotlib.pyplot.subplots()

Este tutorial explica como podemos adicionar subquadrantes a uma figura Matplotlib utilizando o método matplotlib.pyplot.figure.add_subplot() em Python.

Adicionar Subplot a uma figura utilizando o método matplotlib.pyplot.figure.add_subplot()

import matplotlib.pyplot as plt 

fig=plt.figure(figsize=(5,5))

x=[1,2,3,4,5]
y=[2,4,5,1,2]

subplot=fig.add_subplot(1,1,1)
subplot.plot(x, y)

fig.suptitle("Figure with only one Subplot")
plt.show()

Resultado:

Gera uma figura com apenas uma subparcela. A figura pode ser interpretada como um elemento de alto nível contendo todas as subparcelas e outros elementos da figura. Inicializamos primeiro o objecto da figura de tamanho (5,5) e depois adicionamos um subplot à figura utilizando o método add_subplot(). O primeiro argumento ao método add_subplot() representa o número de linhas de subplot na figura, o segundo argumento representa o número de colunas de subplot, e o terceiro argumento representa a posição de um subplot na grelha.

Aqui, fig.add_subplot(1,1,1) irá adicionar um subplot na primeira posição da grelha 1 X 1 na figura. Alternativamente, podemos também utilizar fig.add_subplot(111) para obter o mesmo resultado. O método add_subplot() devolve um objecto axes, que pode ser utilizado para traçar e adicionar detalhes ao subplot adicionado.

import matplotlib.pyplot as plt 

fig=plt.figure(figsize=(8,6))

ax_1=fig.add_subplot(121)
ax_1.text(0.3, 0.5, 'subplot(121)')

ax_2=fig.add_subplot(222)
ax_2.text(0.3, 0.5, 'subplot(222)')

ax_3=fig.add_subplot(224)
ax_3.text(0.3, 0.5, 'subplot(224)')

fig.suptitle("Figure with multiple Subplots")
plt.show()

Resultado:

Aqui, temos uma figura com três subquadrantes. A primeira subparcela é a primeira figura da grelha com uma linha e duas colunas. A segunda subparcela representa a segunda figura da grelha com duas filas e duas colunas. Da mesma forma, a terceira linha representa a subparcela na quarta posição da grelha com duas filas e duas colunas.

Criar figura com subparcelas utilizando Matplotlib.pyplot.subplots()

O Matplotlib.pyplot.subplots() também adiciona subplots à figura. Passamos o número de linhas e colunas como argumento para o método, e o método devolve um objecto figura e um objecto eixos, que pode ser utilizado para manipular a parcela.

import matplotlib.pyplot as plt 

fig,ax=plt.subplots(2,1)

ax[0].text(0.3,0.5,"1st Subplot")
ax[0].set_xticks([])
ax[0].set_yticks([])

ax[1].text(0.3,0.5,"2nd Subplot")
ax[1].set_xticks([])
ax[1].set_yticks([])

fig.suptitle('Figure with 2 subplots',fontsize=16) 
plt.show()

Resultado:

Gera uma figura com duas sub-parcelas numa grelha de duas linhas e uma única coluna utilizando o método subplots(). A primeira subparcela ou os eixos no topo podem ser manipulados com o primeiro elemento da lista ax, e a segunda subparcela ou eixos no fundo podem ser manipulados com o segundo elemento da lista ax.

import matplotlib.pyplot as plt 

x=[1,2,3,4,5]
y1=[i**2 for i in x ]
y2=[4*i+1 for i in x]

fig,ax=plt.subplots(2,1)

ax[0].plot(x,y1)
ax[0].set_xlabel("X")
ax[0].set_ylabel("Y")

ax[1].plot(x,y2)
ax[1].set_xlabel("X")
ax[1].set_ylabel("Y")

plt.tight_layout()
plt.show()

Resultado:

Gera uma figura com dois subquadrantes, traça a função x^2 no subquadrantes superior, e traça a função 4x+1 no subquadrantes inferior.