Как установить один главный заголовок над всеми подсюжетами с помощью Pyplot?

Question

Я использую pyplot. У меня есть 4 сюжета. Как установить один главный заголовок над всеми подсюжетами? title() устанавливает его выше последнего подзаговора.

Answer 1

Использование pyplot.suptitle или 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()

Answer 2

Несколько моментов, которые я считаю полезными при применении этого к своим собственным графикам:

• Я предпочитаю последовательность использования fig.suptitle(title) вместо plt.suptitle(title)
• При использовании fig.tight_layout() заголовок должен быть сдвинут на fig.subplots_adjust(top=0.88)
• См. Ответ ниже о размерах шрифта

Пример кода взят из демонстрации subplots в документах matplotlib и скорректирован с использованием основного заголовка.

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()

Answer 3

Если ваши подсюжеты также имеют заголовки, вам может потребоваться настроить основной размер заголовка:

plt.suptitle("Main Title", size=16)