Cuadrículas de ejes en Python

subplots devuelve un arreglo de ejes que indexas como cualquier otro.

import matplotlib

matplotlib.use("Agg")
import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 2 * np.pi, 100)
figura, ejes = plt.subplots(nrows=2, ncols=2, figsize=(7, 5), sharex=True)

for ax, (nombre, valores) in zip(ejes.ravel(), [
    ("sin", np.sin(x)), ("cos", np.cos(x)),
    ("sin^2", np.sin(x) ** 2), ("amortiguada", np.sin(x) * np.exp(-x / 5)),
]):
    ax.plot(x, valores)
    ax.set_title(nombre, fontsize=9)

figura.tight_layout()
print(ejes.shape)
figura.savefig("grid.png")

Cómo funciona

nrows y ncols dan forma a la cuadrícula.
sharex enlaza los ejes: hacer zoom en uno los mueve todos.
tight_layout evita que las etiquetas choquen.

Palabras clave y builtins usados aquí

as
for
print
zip

El intento, en números

Líneas: 19
Caracteres a escribir: 489
Tokens: 187
Ritmo de tres estrellas: 100 tpm

Al ritmo de tres estrellas de 100 tokens por minuto, este intento toma unos 112 segundos.

Escribe este fragmento

Paso 2 de 4 en Figuras y ejes; paso 2 de 14 en Graficar con matplotlib.

← Anterior Siguiente →