Matplotlib tight_layout() 不考虑图形字幕

Question

如果我在我的 matplotlib 图中添加了一个副标题，它会被副图的标题覆盖。有人知道如何轻松解决这个问题吗？我尝试了tight_layout() 函数，但它只会让事情变得更糟。

例子：

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

Answer 1

您可以使用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()

Answer 2

您可以在代码中轻松更改的一件事是您用于标题的fontsize。但是，我假设您不只是想这样做！

使用fig.subplots_adjust(top=0.85)的一些替代方法：

通常tight_layout() 可以很好地将所有内容放置在合适的位置，以免它们重叠。 tight_layout() 在这种情况下没有帮助的原因是因为 tight_layout() 没有考虑 fig.suptitle()。 GitHub 上有一个关于此的未解决问题：https://github.com/matplotlib/matplotlib/issues/829 [关闭2014 年，由于需要一个完整的几何管理器 - 转移到 https://github.com/matplotlib/matplotlib/issues/1109 ]。

如果您阅读了该主题，您的问题涉及GridSpec 的解决方案。关键是在调用tight_layout 时在图的顶部留出一些空间，使用rect kwarg。对于您的问题，代码变为：

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

结果：

也许GridSpec 对您来说有点矫枉过正，或者您的真正问题将涉及更大画布上的更多子图，或其他复杂情况。一个简单的技巧是只使用annotate() 并将坐标锁定到'figure fraction' 以模仿suptitle。但是，一旦查看输出，您可能需要进行一些更精细的调整。请注意，第二个解决方案不使用tight_layout()。

更简单的解决方案（尽管可能需要微调）

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

结果：

[使用Python2.7.3（64位）和matplotlib1.2.0]

Answer 3

我一直在使用 matplotlib 修剪方法，所以我现在刚刚创建了一个函数来通过 bash 调用 ImageMagick 的 mogrify command 来执行此操作，它运行良好并且可以消除所有额外的空白人物的边缘。这要求您使用 UNIX/Linux，使用 bash shell，并安装了 ImageMagick。

在您的savefig() 调用之后调用它。

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)

Answer 4

另一种简单易用的解决方案是使用 suptitle 调用中的 y 参数调整图中字幕文本的坐标（参见@​​987654321@）：

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

Answer 5

您可以在非常 tight_layout 调用中调整子图几何，如下所示：

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

正如文档 (https://matplotlib.org/users/tight_layout_guide.html) 中所述：

tight_layout() 只考虑刻度标签、轴标签和标题。因此，其他艺术家可能会被剪辑，也可能重叠。

Answer 6

在使用tight_layout 处理非常大的图网格（超过 200 个子图）并在 jupyter 笔记本中进行渲染时，我遇到了类似的问题。我制定了一个快速解决方案，始终将您的suptitle 放置在您的顶部子图上方一定距离处：

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)

对于可变大小的子图，您仍然可以使用此方法获取最顶部子图的顶部，然后手动定义附加量以添加到字幕。

Answer 7

正如其他人所提到的，默认情况下紧凑的布局不考虑字幕。但是，我发现可以使用 bbox_extra_artists 参数将字幕作为边界框传递，应该考虑到这一点：

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

这会强制紧凑的布局计算将suptitle 考虑在内，它看起来与您预期的一样。

Answer 8

紧凑的布局不适用于字幕，但constrained_layout 可以。看到这个问题Improve subplot size/spacing with many subplots in matplotlib

我发现立即添加子图看起来更好，即

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

# then iterating over the axes to fill in the plots

但也可以在图形创建时添加：

fig = plt.figure(constrained_layout=True)

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

注意：为了使我的子图更紧密，我也使用了

fig.subplots_adjust(wspace=0.05)

并且 constrained_layout 不适用于此 :(

Answer 9

唯一对我有用的是修改对 suptitle 的调用：

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

Answer 10

This website 有一个简单的解决方案，其中有一个对我有用的示例。实际为标题留出空间的代码行如下：

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

这是一张证明它对我有用的图片：

Answer 11

截至v3.3 tight_layout 现在支持suptitle：

import matplotlib.pyplot as plt

fig, axs = plt.subplots(1, 3)
for i, ax in enumerate(axs):
    ax.plot([1, 2, 3])
    ax.set_title(f'Axes {i}')

fig.suptitle('suptitle')
fig.tight_layout()