Matplotlib tight_layout () no tiene en cuenta la cifra de los supuestos

Si agrego un subtítulo a mi figura matplotlib, los títulos de la subtrama se superponen. ¿Alguien sabe cómo solucionarlo fácilmente? Probé la tight_layout()función, pero solo empeora las cosas.

Ejemplo:

import numpy as np
import matplotlib.pyplot as plt

f = np.random.random(100)
g = np.random.random(100)
fig = plt.figure()
fig.suptitle('Long Suptitle', fontsize=24)
plt.subplot(121)
plt.plot(f)
plt.title('Very Long Title 1', fontsize=20)
plt.subplot(122)
plt.plot(g)
plt.title('Very Long Title 2', fontsize=20)
plt.tight_layout()
plt.show()

Puede ajustar la geometría de la subtrama en la misma tight_layoutllamada de la siguiente manera:

fig.tight_layout(rect=[0, 0.03, 1, 0.95])

Como se indica en la documentación ( https://matplotlib.org/users/tight_layout_guide.html ):

tight_layout()solo considera las marcas de graduación, las etiquetas de eje y los títulos. Por lo tanto, otros artistas pueden ser recortados y también pueden superponerse.

Puede ajustar manualmente el espaciado usando plt.subplots_adjust(top=0.85):

import numpy as np
import matplotlib.pyplot as plt

f = np.random.random(100)
g = np.random.random(100)
fig = plt.figure()
fig.suptitle('Long Suptitle', fontsize=24)
plt.subplot(121)
plt.plot(f)
plt.title('Very Long Title 1', fontsize=20)
plt.subplot(122)
plt.plot(g)
plt.title('Very Long Title 2', fontsize=20)
plt.subplots_adjust(top=0.85)
plt.show()

Una cosa que podría cambiar en su código muy fácilmente es la fontsizeque está utilizando para los títulos. Sin embargo, ¡voy a suponer que no solo quieres hacer eso!

Algunas alternativas al uso fig.subplots_adjust(top=0.85):

Por tight_layout()lo general, hace un trabajo bastante bueno al colocar todo en buenas ubicaciones para que no se superpongan. La razón por la tight_layout()que no ayuda en este caso es porque tight_layout()no tiene en cuenta fig.suptitle (). Hay un problema abierto sobre esto en GitHub: https://github.com/matplotlib/matplotlib/issues/829 [cerrado en 2014 debido a que requiere un administrador de geometría completo - cambiado a https://github.com/matplotlib/matplotlib / cuestiones / 1109 ].

Si lees el hilo, hay una solución a tu problema GridSpec. La clave es dejar algo de espacio en la parte superior de la figura al llamar tight_layout, usando el rectkwarg. Para su problema, el código se convierte en:

Usando GridSpec

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

f = np.random.random(100)
g = np.random.random(100)

fig = plt.figure(1)
gs1 = gridspec.GridSpec(1, 2)
ax_list = [fig.add_subplot(ss) for ss in gs1]

ax_list[0].plot(f)
ax_list[0].set_title('Very Long Title 1', fontsize=20)

ax_list[1].plot(g)
ax_list[1].set_title('Very Long Title 2', fontsize=20)

fig.suptitle('Long Suptitle', fontsize=24)    

gs1.tight_layout(fig, rect=[0, 0.03, 1, 0.95])  

plt.show()

El resultado:

Tal vez GridSpecsea ​​un poco excesivo para usted, o su problema real implicará muchas más subtramas en un lienzo mucho más grande u otras complicaciones. Un truco simple es usar annotate()y bloquear las coordenadas 'figure fraction'para imitar a suptitle. Sin embargo, es posible que deba hacer algunos ajustes más finos una vez que eche un vistazo a la salida. Tenga en cuenta que esta segunda solución no utilizatight_layout() .

Solución más simple (aunque es posible que deba ajustarse)

fig = plt.figure(2)

ax1 = plt.subplot(121)
ax1.plot(f)
ax1.set_title('Very Long Title 1', fontsize=20)

ax2 = plt.subplot(122)
ax2.plot(g)
ax2.set_title('Very Long Title 2', fontsize=20)

# fig.suptitle('Long Suptitle', fontsize=24)
# Instead, do a hack by annotating the first axes with the desired 
# string and set the positioning to 'figure fraction'.
fig.get_axes()[0].annotate('Long Suptitle', (0.5, 0.95), 
                            xycoords='figure fraction', ha='center', 
                            fontsize=24
                            )
plt.show()

El resultado:

[Utilizando Python2.7.3 (64 bits) y matplotlib1.2.0]

Una solución alternativa y fácil de usar es ajustar las coordenadas del texto del título en la figura usando el argumento y en la llamada del título (ver los documentos ):

import numpy as np
import matplotlib.pyplot as plt

f = np.random.random(100)
g = np.random.random(100)
fig = plt.figure()
fig.suptitle('Long Suptitle', y=1.05, fontsize=24)
plt.subplot(121)
plt.plot(f)
plt.title('Very Long Title 1', fontsize=20)
plt.subplot(122)
plt.plot(g)
plt.title('Very Long Title 2', fontsize=20)
plt.show()

El diseño ajustado no funciona con el título, pero constrained_layoutsí. Ver esta pregunta Mejore el tamaño / espaciado de la subtrama con muchas subtramas en matplotlib

Encontré que agregar las subtramas a la vez se veía mejor, es decir

fig, axs = plt.subplots(rows, cols, constrained_layout=True)

# then iterating over the axes to fill in the plots

Pero también se puede agregar en el momento en que se crea la figura:

fig = plt.figure(constrained_layout=True)

ax1 = fig.add_subplot(cols, rows, 1)
# etc

Nota: para acercar mis subtramas, también estaba usando

fig.subplots_adjust(wspace=0.05)

y restricined_layout no funciona con esto :(

He luchado con el recorte matplotlib métodos, por lo que he ahora acabo de hacer una función para hacer esto a través de una bashllamada a ImageMagick's comando mogrify , que funciona bien y se pone todo el espacio en blanco adicionales fuera del borde de la figura. Esto requiere que esté utilizando UNIX / Linux, esté utilizando el bashshell y lo haya ImageMagickinstalado.

Simplemente envíe una llamada a esto después de su savefig()llamada.

def autocrop_img(filename):
    '''Call ImageMagick mogrify from bash to autocrop image'''
    import subprocess
    import os

    cwd, img_name = os.path.split(filename)

    bashcmd = 'mogrify -trim %s' % img_name
    process = subprocess.Popen(bashcmd.split(), stdout=subprocess.PIPE, cwd=cwd)

Como lo mencionaron otros, por defecto, el diseño ajustado no tiene en cuenta el título. Sin embargo, he descubierto que es posible usar el bbox_extra_artistsargumento para pasar el supítulo como un cuadro delimitador que debe tenerse en cuenta:

st = fig.suptitle("My Super Title")
plt.savefig("figure.png", bbox_extra_artists=[st], bbox_inches='tight')

Esto obliga al cálculo de diseño ajustado a tener suptitleen cuenta, y se ve como es de esperar.

Lo único que funcionó para mí fue modificar la llamada al suptitle:

fig.suptitle("title", y=.995)

Tuve un problema similar que surgió cuando se usaba tight_layoutpara una cuadrícula de tramas muy grande (más de 200 subtramas) y se renderizaba en un cuaderno jupyter. Hice una solución rápida que siempre te coloca suptitlea cierta distancia por encima de tu subtrama superior:

import matplotlib.pyplot as plt

n_rows = 50
n_col = 4
fig, axs = plt.subplots(n_rows, n_cols)

#make plots ...

# define y position of suptitle to be ~20% of a row above the top row
y_title_pos = axs[0][0].get_position().get_points()[1][1]+(1/n_rows)*0.2
fig.suptitle('My Sup Title', y=y_title_pos)

Para las subtramas de tamaño variable, aún puede utilizar este método para obtener la parte superior de la subtrama superior, luego definir manualmente una cantidad adicional para agregar al título.