¿Por qué muchos ejemplos usan `fig, ax = plt.subplots ()` en Matplotlib / pyplot / python

Estoy aprendiendo a usar matplotlibal estudiar ejemplos, y muchos ejemplos parecen incluir una línea como la siguiente antes de crear una sola trama ...

fig, ax = plt.subplots()

Aquí hay unos ejemplos...

• Modificar texto de etiqueta de marca
• http://matplotlib.org/examples/pylab_examples/boxplot_demo2.html

Veo que esta función se usa mucho, a pesar de que el ejemplo solo intenta crear un solo gráfico. ¿Hay alguna otra ventaja? La demostración oficial de subplots()también se usa f, ax = subplotsal crear un solo gráfico, y solo hace referencia a axe después de eso. Este es el código que usan.

# Just a figure and one subplot
f, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('Simple plot')

plt.subplots()es una función que devuelve una tupla que contiene una figura y eje (s) objeto (s). Por lo tanto, al usar fig, ax = plt.subplots(), descomprime esta tupla en las variables figy ax. Tener figes útil si desea cambiar los atributos de nivel de figura o guardar la figura como un archivo de imagen más tarde (por ejemplo, con fig.savefig('yourfilename.png')). Ciertamente no tiene que usar el objeto de figura devuelto, pero muchas personas lo usan más tarde, por lo que es común verlo. Además, todos los objetos de los ejes (los objetos que tienen métodos de trazado) tienen un objeto de figura principal de todos modos, por lo tanto:

fig, ax = plt.subplots()

es más conciso que esto:

fig = plt.figure()
ax = fig.add_subplot(111)

Solo un suplemento aquí.

La siguiente pregunta es: ¿qué sucede si quiero más subtramas en la figura?

Como se menciona en el Doc, podemos usar fig = plt.subplots(nrows=2, ncols=2) para establecer un grupo de subtramas con cuadrícula (2,2) en un objeto de figura.

Entonces, como sabemos, el fig, ax = plt.subplots()resultado es una tupla, intentemos fig, ax1, ax2, ax3, ax4 = plt.subplots(nrows=2, ncols=2)primero.

ValueError: not enough values to unpack (expected 4, got 2)

Provoca un error, pero no te preocupes, porque ahora vemos que plt.subplots() realidad devuelve una tupla con dos elementos. El primero debe ser un objeto de figura y el otro debe ser un grupo de objetos de subtramas.

Entonces intentemos esto de nuevo:

fig, [[ax1, ax2], [ax3, ax4]] = plt.subplots(nrows=2, ncols=2)

y verifique el tipo:

type(fig) #<class 'matplotlib.figure.Figure'>
type(ax1) #<class 'matplotlib.axes._subplots.AxesSubplot'>

Por supuesto, si usa parámetros como (nrows = 1, ncols = 4), entonces el formato debe ser:

fig, [ax1, ax2, ax3, ax4] = plt.subplots(nrows=1, ncols=4)

Así que solo recuerde mantener la construcción de la lista igual que la cuadrícula de subtramas que establecemos en la figura.

Espero que esto sea útil para ti.

Como complemento de la pregunta y las respuestas anteriores, también hay una diferencia importante entre plt.subplots()y plt.subplot(), observe la falta 's'al final.

Uno puede usar plt.subplots()para hacer todas sus subtramas a la vez y devuelve la figura y los ejes (plural del eje) de las subtramas como una tupla. Una figura puede entenderse como un lienzo donde pintas tu boceto.

# create a subplot with 2 rows and 1 columns
fig, ax = plt.subplots(2,1)

Mientras que, puede usar plt.subplot()si desea agregar las subtramas por separado. Solo devuelve el eje de una subtrama.

fig = plt.figure() # create the canvas for plotting
ax1 = plt.subplot(2,1,1) 
# (2,1,1) indicates total number of rows, columns, and figure number respectively
ax2 = plt.subplot(2,1,2)

Sin embargo, plt.subplots()se prefiere porque le brinda opciones más fáciles de personalizar directamente su figura completa

# for example, sharing x-axis, y-axis for all subplots can be specified at once
fig, ax = plt.subplots(2,2, sharex=True, sharey=True)

mientras que, con plt.subplot(), habrá que especificar individualmente para cada eje que puede volverse engorroso.

Además de las respuestas anteriores, puede verificar el tipo de objeto type(plt.subplots())que usa que devuelve una tupla, por otro lado, type(plt.subplot())devuelve el matplotlib.axes._subplots.AxesSubplotque no puede desempaquetar.