La API pyplot de Matplotlib tiene una función de conveniencia llamada subplots () que actúa como un contenedor de utilidad y ayuda a crear diseños comunes de subplots, incluido el objeto de figura adjunto, en una sola llamada.
Plt.subplots(nrows, ncols)
Los dos argumentos enteros de esta función especifican el número de filas y columnas de la cuadrícula de la subparcela. La función devuelve un objeto de figura y una tupla que contiene objetos de ejes iguales a nrows * ncols. Cada objeto de los ejes es accesible por su índice. Aquí creamos una subparcela de 2 filas por 2 columnas y mostramos 4 gráficas diferentes en cada subparcela.
import matplotlib.pyplot as plt
fig,a = plt.subplots(2,2)
import numpy as np
x = np.arange(1,5)
a[0][0].plot(x,x*x)
a[0][0].set_title('square')
a[0][1].plot(x,np.sqrt(x))
a[0][1].set_title('square root')
a[1][0].plot(x,np.exp(x))
a[1][0].set_title('exp')
a[1][1].plot(x,np.log10(x))
a[1][1].set_title('log')
plt.show()
La línea de código anterior genera la siguiente salida: