如何在Python的Matplotlib中保存图形

Question

我想在Matplotlib中保存我的图形文件，但是当我运行保存图像的命令时，它没有报错，但我无法看到该文件。

plt.savefig('Traveling Salesmen Graph.png')

- Spidey Zac

---

请在您当前的工作目录中查找它。 - Madhoolika

plt.savefig() goes before plt.show() - Trenton McKinney

将绘图保存为图像文件而不是使用Matplotlib显示。 - Trenton McKinney

@TrentonMcKinney，我没有plt.show()，我只有plt.plot()。 - Spidey Zac

请提供一个最小可复现示例（例如代码、数据、错误）作为文本。 - Trenton McKinney

@SpideyZac 可能有助于将文件保存到特定位置并在那里检查。当我运行您的命令时，图形保存到工作目录中。 - Madhoolika

Answer 1

pyplot会跟踪"当前图形"，在库中调用需要图形的函数时都会在其上操作。但是您也可以通过在图形对象上调用savefig()来更加明确。

以下是来自https://pythonspot.com/matplotlib-save-figure-to-image-file/的示例:

import matplotlib
import matplotlib.pyplot as plt
import numpy as np

y = [2,4,6,8,10,12,14,16,18,20]
x = np.arange(10)
fig = plt.figure()
ax = plt.subplot(111)
ax.plot(x, y, label='$y = numbers')
plt.title('Legend inside')
ax.legend()
#plt.show()

fig.savefig('plot.png')

这样明确地表达应该解决你的问题。

有关操作“当前图形”的pyplot函数的参考，请参见：https://matplotlib.org/3.2.1/api/_as-gen/matplotlib.pyplot.html

- Mark McElroy

---

plt.savefig是一个非常重要的函数。这并不是事实正确的。 - Mad Physicist

@MadPhysicist 您是正确的，我已经修改了答案。 - Mark McElroy