在matplotlib中创建两个完全独立的图并在它们之间来回传递

时间:2014-02-02 17:28:24

标签: python matplotlib

我想在python脚本中创建两个独立的matplotlib图,并且可能在它们之间来回跳转,因为我在各个图中添加了行,注释等(例如,我可能会调用一个添加的函数)线条到两个图,然后是另一个添加注释的函数)。

我希望通过使用matplotlib示例,我能够找到一些有效的解决方案,但我想知道这样做的首选和最干净的方法是什么。我倾向于对何时应该做

这样的事情感到困惑
fig,ax=plt.subplots()

当我应该做的事情:

fig=plt.figure()

此外,我应该如何在地块之间来回切换。如果我做了类似

的事情
fig1,ax1=plt.subplots()
fig2,ax2=plt.subplots()

然后我可以通过执行以下操作来参考这些图:

ax1.plt.plot([some stuff])
ax2.plt.plot([otherstuff]

?我问这个是因为经常在matplotlib示例中他们在调用plt.subplot()之后没有引用这样的图,而是调用像

这样的命令
plt.plot([stuff])

假设他们没有指定ax1或ax2并不重要,因为示例中只有一个图。最后,我想使用类似

之类的内容将两个图表保存到文件中
plt.savefig(....)

虽然我再次需要能够独立地参考这两个图。那么实现这个的正确的方式是什么?

1 个答案:

答案 0 :(得分:4)

如果您希望能够编写明确将命令应用于不同轴的代码,则需要使用object oriented interface

实际上,前两个示例都使用此接口。不同之处在于plt.subplots()将同时创建一个图形对象和一个轴网格,而plt.figure()只会创建该图形。

图形对象具有在其中创建轴的方法。所以,这两个代码块是等价的:

fig, ax = plt.subplots()

fig = plt.figure()
ax = fig.add_subplot(111)

通常,当您希望图中的多个轴不遵循常规网格时,后一种方法才会更有用。所以,你可以这样做:

fig = plt.figure()
ax = fig.add_axes([.1, .1, .2, .8])

将在图的左侧添加一个高轴对象。


接下来,如何绘制多个轴?

subplots函数有两个位置参数,指定网格中的行数和列数(默认为(1, 1)。所以如果你想要两个轴并排,你会做

fig, axes = plt.subplots(1, 2)

现在axes将是一个用Axes对象填充的(1, 2)对象数组。对于小网格来说,使用Python的元组解包并获得对象的直接引用通常更方便:

fig, (ax1, ax2) = plt.subplots(1, 2)

现在,您如何处理这些对象,以及它们与MATLAB样式的过程接口之间的关系是什么?

pyplot命名空间中的大多数函数也作为Figure或Axes对象上的方法存在。 Matplotlib(和MATLAB)具有“当前”图形和轴的概念。当您调用类似plt.plot的函数时,它会在当前轴(通常是最近创建的轴)上绘制。当您调用类似plt.savefig的函数时,它会保存当前数字。

对于简单的任务,这比使用面向对象的接口更直接,更容易。但是,当您开始制作更复杂的图表时,例如一个轴网格,每个网格有多个层(可能是一个散点图和一个回归线),能够构建围绕你正在做什么的代码,而不是 你在哪里做有很大的优势。通常,以面向对象的方式编写的绘图代码将比以过程方式编写的代码更好地扩展。