Agregar subplot a una figura en Matplotlib

1. Agregar subplot a una figura usando el método matplotlib.pyplot.figure.add_subplot()
2. Crear una figura con subplots usando Matplotlib.pyplot.subplots()

Este tutorial explica cómo podemos agregar subplots a una figura Matplotlib utilizando el método matplotlib.pyplot.figure.add_subplot() en Python.

Agregar subplot a una figura usando el 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:

Genera una figura con una sola subplot. La figura puede interpretarse como un elemento de alto nivel que contiene todas las subplots y otros elementos de la figura. Primero inicializamos la figura objeto de tamaño (5,5) y luego agregamos una subplot a la figura usando el método add_subplot(). El primer argumento del método add_subplot() representa el número de filas de la subplot en la figura, el segundo argumento representa el número de columnas de la subplot, y el tercer argumento representa la posición de una subplot en la cuadrícula.

Aquí, fig.add_subplot(1,1,1) agregará una subplot en la primera posición de la cuadrícula 1 X 1 de la figura. Alternativamente, también podemos usar fig.add_subplot(111) para lograr el mismo resultado. El método add_subplot() devuelve un objeto axes, que puede ser usado para trazar y añadir detalles a la subplot añadida.

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:

Aquí, tenemos una figura con tres subplots. La primera subplot es la primera figura de la cuadrícula con una fila y dos columnas. La segunda subplot representa la segunda figura en la cuadrícula con dos filas y dos columnas. De manera similar, la tercera fila representa la subplot en la cuarta posición de la cuadrícula con dos filas y dos columnas.

Crear una figura con subplots usando Matplotlib.pyplot.subplots()

El Matplotlib.pyplot.subplots() también añade subplots a la figura. Le pasamos el número de filas y columnas como argumento al método, y el método devuelve un objeto de la figura y un objeto de los ejes, que pueden ser usados para manipular el gráfico.

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:

Genera una figura con dos subplots en una cuadrícula de dos filas y una sola columna usando el método subplots(). La primera subplot o los ejes de la parte superior pueden ser manipulados con el primer elemento de la lista de los ax, y la segunda subplot o los ejes de la parte inferior pueden ser manipulados con el segundo elemento de la lista de los 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:

Genera una figura con dos subplots, traza la función x^2 en la subplot superior, y traza la función 4x+1 en la subplot inferior.