matplotlib：隐藏子图并用其他子图填充空间

Question

我有一个包含三个垂直排列的子图的图形。一旦我点击图中，我希望隐藏第二个子图ax2，另一个图填充空间。再次点击图中应恢复原始图和布局。

隐藏子情节ax2不是问题，但如何重新排列其他子图的位置？

我尝试使用GridSpec和set_position方法创建新的set_subplotspec，但没有任何结果。我确定我在这里遗漏了一些东西，任何帮助都会受到赞赏。

这是我的代码：

import matplotlib.pyplot as plt
from matplotlib import gridspec

fig = plt.figure()

gs = gridspec.GridSpec(3, 1, height_ratios=[5, 2, 1])
ax1 = fig.add_subplot(gs[0])
ax2 = fig.add_subplot(gs[1], sharex=ax1)
ax3 = fig.add_subplot(gs[2], sharex=ax2)

visible = True

def toggle_ax2(event):
    global visible
    visible = not visible
    ax2.set_visible(visible)
    plt.draw()

fig.canvas.mpl_connect('button_press_event', toggle_ax2)
plt.show()

Answer 1

您可以定义两个不同的GridSpec。一个将有3个子图，另一个2.根据中轴的可见性，您可以更改其他两个轴的位置以服从第一个或第二个GridSpec。
（不需要任何虚拟数字，就像其他答案可能暗示的那样。）

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

fig = plt.figure()

gs = gridspec.GridSpec(3, 1, height_ratios=[5, 2, 1], hspace=0.3)
gs2 = gridspec.GridSpec(2,1, height_ratios=[5,3])

ax1 = fig.add_subplot(gs[0])
ax2 = fig.add_subplot(gs[1], sharex=ax1)
ax3 = fig.add_subplot(gs[2], sharex=ax2)

ax1.plot([1,2,3], [1,2,3], color="crimson")
ax2.plot([1,2,3], [2,3,1], color="darkorange")
ax3.plot([1,2,3], [3,2,1], color="limegreen")

visible = True

def toggle_ax2(event):
    global visible
    visible = not visible
    ax2.set_visible(visible)
    if visible:
        ax1.set_position(gs[0].get_position(fig))
        ax3.set_position(gs[2].get_position(fig))
    else:
        ax1.set_position(gs2[0].get_position(fig))
        ax3.set_position(gs2[1].get_position(fig))

    plt.draw()

fig.canvas.mpl_connect('button_press_event', toggle_ax2)
plt.show()

左：原始;右：点击

后