在Python中将绘图保存到png文件中

Question

我使用matplotlib.pyplot在Python中创建了一个简单的图。代码在这里：

import matplotlib.pyplot as plt
import bumpy as np
@np.vectorize
def f(x):
    return x
x = np.linspace(0, 10, 1000)
plt.plot(x, f(x), 'r')
plt.show()

现在我想将该图保存到png文件中，我将使用plt.savefig（）。我试着做这样的事情：

import matplotlib.pyplot as plt
import bumpy as np
@np.vectorize
def f(x):
    return x
x = np.linspace(0, 10, 1000)
plt.plot(x, f(x), 'r')    
plt.show(block = False)
plt.savefig("plot.png", format = 'png')

没有错误，但遗憾的是它不起作用。如何正确使用plt.savefig（）？