¿Cómo configurar un solo título principal sobre todas las subtramas con Pyplot?

219

Estoy usando pyplot. Tengo 4 subtramas. ¿Cómo establecer un solo título principal sobre todas las subtramas? title()lo establece sobre la última subtrama.

Jakub M.
fuente

Respuestas:

283

Usar pyplot.suptitleo Figure.suptitle:

import matplotlib.pyplot as plt
import numpy as np

fig=plt.figure()
data=np.arange(900).reshape((30,30))
for i in range(1,5):
    ax=fig.add_subplot(2,2,i)        
    ax.imshow(data)

fig.suptitle('Main title') # or plt.suptitle('Main title')
plt.show()

ingrese la descripción de la imagen aquí

unutbu
fuente
1
Funciona con suptitle. Aún así, vi tu "desvergonzado truco". :)
Jakub M.
66
Tenga en cuenta que es plt.suptitle()y no plt.subtitle(). ¡No me di cuenta de esto al principio y obtuve un error desagradable! : D
Dataman
127

Algunos puntos que encuentro útiles al aplicar esto a mis propias parcelas:

  • Prefiero la consistencia de usar en fig.suptitle(title)lugar deplt.suptitle(title)
  • Cuando se usa fig.tight_layout()el título, se debe cambiar config.subplots_adjust(top=0.88)
  • Vea la respuesta a continuación sobre los tamaños de fuente

Código de ejemplo tomado de la demostración de subtramas en documentos de matplotlib y ajustado con un título maestro.

Una bonita parcela 4x4

import matplotlib.pyplot as plt
import numpy as np

# Simple data to display in various forms
x = np.linspace(0, 2 * np.pi, 400)
y = np.sin(x ** 2)

fig, axarr = plt.subplots(2, 2)
fig.suptitle("This Main Title is Nicely Formatted", fontsize=16)

axarr[0, 0].plot(x, y)
axarr[0, 0].set_title('Axis [0,0] Subtitle')
axarr[0, 1].scatter(x, y)
axarr[0, 1].set_title('Axis [0,1] Subtitle')
axarr[1, 0].plot(x, y ** 2)
axarr[1, 0].set_title('Axis [1,0] Subtitle')
axarr[1, 1].scatter(x, y ** 2)
axarr[1, 1].set_title('Axis [1,1] Subtitle')

# # Fine-tune figure; hide x ticks for top plots and y ticks for right plots
plt.setp([a.get_xticklabels() for a in axarr[0, :]], visible=False)
plt.setp([a.get_yticklabels() for a in axarr[:, 1]], visible=False)

# Tight layout often produces nice results
# but requires the title to be spaced accordingly
fig.tight_layout()
fig.subplots_adjust(top=0.88)

plt.show()
Alexander McFarlane
fuente
2
Simplemente agregar figure.suptitle()no es suficiente ya que los títulos de subtramas se mezclarán con el suptitile, fig.subplots_adjust(top=0.88)es bueno.
GoingMyWay
43

Si sus subtramas también tienen títulos, es posible que deba ajustar el tamaño del título principal:

plt.suptitle("Main Title", size=16)
pentandroso
fuente
Cambiar el tamaño de la fuente lo hace mucho mejor. ¡Gracias!
Wok
8
En python 2.7 es de fuente en lugar de tamaño . plt.suptitle("Main Title", fontsize=16)
Temak