使用 Python 的 matplotlib 选择要在屏幕上显示的图形以及要保存到文件中的图形

Question

我想使用matplotlib.pyplot 在 Python 中创建不同的图形。然后我想将其中一些保存到一个文件中，而其他的应该使用show() 命令显示在屏幕上。

但是，show() 显示所有创建的人物。在创建不想在屏幕上显示的绘图后，我可以通过调用close() 来避免这种情况，如下面的代码：

import matplotlib.pyplot as plt

y1 = [4, 2, 7, 3]
y2 = [-7, 0, -1, -3]

plt.figure()
plt.plot(y1)
plt.savefig('figure1.png')
plt.close()

plt.figure()
plt.plot(y2)

plt.show()
plt.close('all')

这会保存第一个图形并显示第二个图形。但是，我收到一条错误消息：

无法调用“事件”命令：应用程序在执行时已被销毁

是否可以以更优雅的方式选择要显示的图形？

另外，第一个figure() 命令是多余的吗？给不给好像没什么区别。

非常感谢。

Answer 1

更好的方法是使用plt.clf() 而不是plt.close()。 此外plt.figure() 会创建一个新图表，而您可以使用plt.clf() 清除之前的图表：

import matplotlib.pyplot as plt

y1 = [4, 2, 7, 3]
y2 = [-7, 0, -1, -3]

plt.figure()
plt.plot(y1)
plt.savefig('figure1.png')
plt.clf()

plt.plot(y2)

plt.show()
plt.clf()

此代码不会产生错误或警告，例如无法调用“事件”命令...

Answer 2

一般来说，你可以直接关闭图。举个简单的例子：

import matplotlib.pyplot as plt

fig1 = plt.figure()
plt.plot(range(10), 'ro-')
plt.title('This figure will be saved but not shown')
fig1.savefig('fig1.png')
plt.close(fig1)

fig2 = plt.figure()
plt.plot(range(10), 'bo')
plt.title('This figure will be shown')

plt.show()

至于第一个plt.figure() 调用是否是多余的，这取决于你在做什么。通常，您希望保留它返回的图形对象并使用它，而不是使用 matplotlib 的 matlab-ish 状态机接口。

当您制作更复杂的绘图时，通常值得多写一行代码来执行以下操作：

fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.plot(range(10))

优点是您不必担心哪个图形或轴是“活动的”，您只需引用特定的轴或图形对象。