Xlabel / ylabel común para subplots matplotlib

Tengo la siguiente trama:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

y ahora me gustaría dar a este gráfico etiquetas comunes del eje xy etiquetas del eje y. Con "común", quiero decir que debería haber una etiqueta grande del eje x debajo de la cuadrícula completa de subtramas, y una etiqueta grande del eje y a la derecha. No puedo encontrar nada sobre esto en la documentación plt.subplots, y mis google sugieren que necesito hacer un gran plt.subplot(111)comienzo, pero ¿cómo pongo mis subtramas 5 * 2 en ese uso plt.subplots?

Esto se parece a lo que realmente quieres. Aplica el mismo enfoque de esta respuesta a su caso específico:

import matplotlib.pyplot as plt

fig, ax = plt.subplots(nrows=3, ncols=3, sharex=True, sharey=True, figsize=(6, 6))

fig.text(0.5, 0.04, 'common X', ha='center')
fig.text(0.04, 0.5, 'common Y', va='center', rotation='vertical')

Como lo considero relevante y lo suficientemente elegante (no es necesario especificar coordenadas para colocar texto), copio (con una ligera adaptación) una respuesta a otra pregunta relacionada .

import matplotlib.pyplot as plt
fig, axes = plt.subplots(5, 2, sharex=True, sharey=True, figsize=(6,15))
# add a big axis, hide frame
fig.add_subplot(111, frameon=False)
# hide tick and tick label of the big axis
plt.tick_params(labelcolor='none', top=False, bottom=False, left=False, right=False)
plt.xlabel("common X")
plt.ylabel("common Y")

Esto da como resultado lo siguiente (con matplotlib versión 2.2.0):

Sin sharex=True, sharey=Trueque consigas:

Con él deberías hacerlo más agradable:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))

plt.tight_layout()

Pero si desea agregar etiquetas adicionales, debe agregarlas solo a los gráficos de borde:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))
        if i == len(axes2d) - 1:
            cell.set_xlabel("noise column: {0:d}".format(j + 1))
        if j == 0:
            cell.set_ylabel("noise row: {0:d}".format(i + 1))

plt.tight_layout()

Agregar etiqueta para cada gráfico lo estropearía (tal vez haya una manera de detectar automáticamente etiquetas repetidas, pero no estoy al tanto de una).

Desde el comando:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

usó devuelve una tupla que consiste en la figura y una lista de las instancias de los ejes, ya es suficiente hacer algo como (tenga en cuenta que he cambiado fig,axa fig,axes):

fig,axes = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

for ax in axes:
    ax.set_xlabel('Common x-label')
    ax.set_ylabel('Common y-label')

Si desea cambiar algunos detalles en una subtrama específica, puede acceder a través de axes[i]donde iitera sobre sus subtramas.

También podría ser muy útil incluir un

fig.tight_layout()

al final del archivo, antes del plt.show(), para evitar la superposición de etiquetas.

Se verá mejor si reserva espacio para las etiquetas comunes haciendo etiquetas invisibles para la subtrama en la esquina inferior izquierda. También es bueno pasar el tamaño de fuente de rcParams. De esta manera, las etiquetas comunes cambiarán de tamaño con la configuración de su rc, y los ejes también se ajustarán para dejar espacio para las etiquetas comunes.

fig_size = [8, 6]
fig, ax = plt.subplots(5, 2, sharex=True, sharey=True, figsize=fig_size)
# Reserve space for axis labels
ax[-1, 0].set_xlabel('.', color=(0, 0, 0, 0))
ax[-1, 0].set_ylabel('.', color=(0, 0, 0, 0))
# Make common axis labels
fig.text(0.5, 0.04, 'common X', va='center', ha='center', fontsize=rcParams['axes.labelsize'])
fig.text(0.04, 0.5, 'common Y', va='center', ha='center', rotation='vertical', fontsize=rcParams['axes.labelsize'])

Me encontré con un problema similar al trazar una cuadrícula de gráficos. Los gráficos constaban de dos partes (superior e inferior). Se suponía que la etiqueta y estaba centrada sobre ambas partes.

No quería usar una solución que dependiera de conocer la posición en la figura externa (como fig.text ()), por lo que manipulé la posición y de la función set_ylabel (). Por lo general, es 0.5, el medio del gráfico al que se agrega. Como el relleno entre las partes (hspace) en mi código era cero, pude calcular el medio de las dos partes en relación con la parte superior.

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

# Create outer and inner grid
outerGrid = gridspec.GridSpec(2, 3, width_ratios=[1,1,1], height_ratios=[1,1])
somePlot = gridspec.GridSpecFromSubplotSpec(2, 1,
               subplot_spec=outerGrid[3], height_ratios=[1,3], hspace = 0)

# Add two partial plots
partA = plt.subplot(somePlot[0])
partB = plt.subplot(somePlot[1])

# No x-ticks for the upper plot
plt.setp(partA.get_xticklabels(), visible=False)

# The center is (height(top)-height(bottom))/(2*height(top))
# Simplified to 0.5 - height(bottom)/(2*height(top))
mid = 0.5-somePlot.get_height_ratios()[1]/(2.*somePlot.get_height_ratios()[0])
# Place the y-label
partA.set_ylabel('shared label', y = mid)

plt.show()

imagen

Desventajas:

• La distancia horizontal al gráfico se basa en la parte superior, las marcas inferiores pueden extenderse a la etiqueta.

• La fórmula no tiene en cuenta el espacio entre las partes.

• Lanza una excepción cuando la altura de la parte superior es 0.

Probablemente haya una solución general que tenga en cuenta el relleno entre las figuras.

Actualizar:

Esta característica ahora es parte del paquete proplot matplotlib que lancé recientemente en pypi. Por defecto, cuando crea figuras, las etiquetas se "comparten" entre los ejes.

Respuesta original:

Descubrí un método más robusto:

Si conoce los bottomy topkwargs que entraron en una GridSpecinicialización, o conoce las posiciones de los bordes de sus ejes en Figurecoordenadas , también puede especificar la posición de la etiqueta en Figurecoordenadas con un poco de magia de "transformación". Por ejemplo:

import matplotlib.transforms as mtransforms
bottom, top = .1, .9
f, a = plt.subplots(nrows=2, ncols=1, bottom=bottom, top=top)
avepos = (bottom+top)/2
a[0].yaxis.label.set_transform(mtransforms.blended_transform_factory(
       mtransforms.IdentityTransform(), f.transFigure # specify x, y transform
       )) # changed from default blend (IdentityTransform(), a[0].transAxes)
a[0].yaxis.label.set_position((0, avepos))
a[0].set_ylabel('Hello, world!')

... y debería ver que la etiqueta todavía se ajusta adecuadamente de izquierda a derecha para evitar que se superpongan con las etiquetas, como de costumbre, pero ahora se ajustará para estar siempre exactamente entre las subtramas deseadas.

Además, si ni siquiera lo usa set_position, la etiqueta aparecerá por defecto exactamente a la mitad de la figura . Supongo que esto se debe a que cuando la etiqueta finalmente se dibuja, matplotlibusa 0.5 para la ycoordenada sin verificar si la transformación de coordenadas subyacente ha cambiado.