如何在Matplotlib中单独和在子图中绘制相同的图？

Question

我正在用Python编写一个生成许多图形的程序。其中一些既独立又与其他图形相比都很有趣。生成这些图表很昂贵（就运行时而言），我不想多次生成它们。有没有办法生成一次情节，并将其作为子情节的一部分？

我基本上是在寻找替代方案：

#generate standalone graphs
pylab.figure()
generate_plot0()
pylab.figure()
generate_plot1()

#generate subplot
pylab.figure()
subplot(121)
generate_plot0()
subplot(122)
generate_plot1()

但是没有两次调用generate_plot0()和generate_plot1()。

有没有好办法呢？

Answer 1

一般来说，matplotlib艺术家不能在多个轴上，并且轴不能在多个图中。 （在某些情况下，你可以破坏其中一些规则，但它一般不起作用。）

因此，简短的回答是否定的。

但是，您可能会考虑以下内容。您可以将该图表作为子图，然后绑定一个单击/按键/任意隐藏所有其他子图并使所选轴暂时填满整个图。

作为一个简单的例子：

import numpy as np
import matplotlib.pyplot as plt

def main():
    subplots = ZoomingSubplots(2, 2)
    colors = ['red', 'green', 'blue', 'cyan']
    for ax, color in zip(subplots.axes.flat, colors):
        data = (np.random.random(200) - 0.5).cumsum()
        ax.plot(data, color=color)
    subplots.fig.suptitle('Click on an axes to make it fill the figure.\n'
                 'Click again to restore it to its original position')
    plt.show()

class ZoomingSubplots(object):
    def __init__(self, *args, **kwargs):
        """All parameters passed on to 'subplots`."""
        self.fig, self.axes = plt.subplots(*args, **kwargs)
        self._zoomed = False
        self.fig.canvas.mpl_connect('button_press_event', self.on_click)

    def zoom(self, selected_ax):
        for ax in self.axes.flat:
            ax.set_visible(False)
        self._original_size = selected_ax.get_position()
        selected_ax.set_position([0.125, 0.1, 0.775, 0.8])
        selected_ax.set_visible(True)
        self._zoomed = True

    def unzoom(self, selected_ax):
        selected_ax.set_position(self._original_size)
        for ax in self.axes.flat:
            ax.set_visible(True)
        self._zoomed = False

    def on_click(self, event):
        if event.inaxes is None:
            return
        if self._zoomed:
            self.unzoom(event.inaxes)
        else:
            self.zoom(event.inaxes)
        self.fig.canvas.draw()

if __name__ == '__main__':
    main()

初始状态

点击子图

后