如何以编程方式告诉matplotlib绘图窗口保存和关闭

Question

此代码段将打开“打印”窗口，并一直保持打开状态，直到我单击关闭它。如何让Matplotlib保存图形并关闭窗口

# From https://matplotlib.org/tutorials/introductory/pyplot.html#sphx-glr-tutorials-introductory-pyplot-py
import matplotlib.pyplot as plt
plt.plot([1, 2, 3, 4])
plt.ylabel('some numbers')
plt.show()  # This code blocks until the plot window is closed.

Answer 1

如果要保存图形而不是显示图形，只需使用plt.savefig而不是plt.show()：

import matplotlib.pyplot as plt
plt.plot([1, 2, 3, 4])
plt.ylabel('some numbers')

plt.savefig('fig.png')

Answer 2

Matplotlib允许您在不打开窗口的情况下轻松保存图形，请使用以下命令：

import matplotlib.pyplot as plt
plt.plot([1, 2, 3, 4])
plt.ylabel('some numbers')
plt.savefig('foo.png')