Question

在我的代码中，用户应该能够更改图中的子图数量。因此，首先有两个子图：

我使用以下代码：

ax1 = figure.add_sublots(2,1,1)
ax2 = figure.add_sublots(2,1,2)

如果按下加号按钮，则应添加一个子图：

我应该怎么做？是否有类似的命令

ax1.change_subplot(3,1,1)
ax2.change_subplot(3,1,2)
ax3 = figure.add_sublots(3,1,3)

还是我必须删除所有子图并重新绘制它们（我想避免这种情况）？

Answer 1

这里是一种选择。您可以为每个要显示的子图数量创建一个新的GridSpec，并根据该gridspec设置轴的位置。

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec
from matplotlib.widgets import Button

class VariableGrid():
    def __init__(self,fig):
        self.fig = fig
        self.axes = []
        self.gs = None
        self.n = 0

    def update(self):
        if self.n > 0:
            for i,ax in zip(range(self.n), self.axes):
                ax.set_position(self.gs[i-1].get_position(self.fig))
                ax.set_visible(True)

            for j in range(len(self.axes),self.n,-1 ):
                print(self.n, j)
                self.axes[j-1].set_visible(False)
        else:
            for ax in self.axes:
                ax.set_visible(False)
        self.fig.canvas.draw_idle()


    def add(self, evt=None):
        self.n += 1
        self.gs= GridSpec(self.n,1)
        if self.n > len(self.axes):
            ax = fig.add_subplot(self.gs[self.n-1])
            self.axes.append(ax)
        self.update()

    def sub(self, evt=None):
        self.n = max(self.n-1,0)
        if self.n > 0:
            self.gs= GridSpec(self.n,1)
        self.update()


fig = plt.figure()

btn_ax1 = fig.add_axes([.8,.02,.05,.05])
btn_ax2 = fig.add_axes([.855,.02,.05,.05])
button_add =Button(btn_ax1, "+")
button_sub =Button(btn_ax2, "-")


grid = VariableGrid(fig)
button_add.on_clicked(grid.add)
button_sub.on_clicked(grid.sub)

plt.show()

Answer 2

我正在寻找的命令是：

ax1.change_geometry(3,1,1)

使用此命令可以重新排列子位。我在这里找到了这个解决方案： Dynamically add subplots in matplotlib with more than one column

Matplotlib-用户更改子图的数量

2 个答案: