matplotlib画图保存

1. import numpy as np
2. import matplotlib.pyplot as plt
3. xData = np.arange(0, 10, 1)
4. yData1 = xData.__pow__(2.0)
5. yData2 = np.arange(15, 61, 5)
6. plt.figure(num=1, figsize=(8, 6))
7. plt.title('Plot 1', size=14)
8. plt.xlabel('x-axis', size=14)
9. plt.ylabel('y-axis', size=14)
10. plt.plot(xData, yData1, color='b', linestyle='--', marker='o', label='y1 data')
11. plt.plot(xData, yData2, color='r', linestyle='-', label='y2 data')
12. plt.legend(loc='upper left')
13. plt.savefig('images/plot1.png', format='png')

fig = plt.figure(0) # 新图 0
plt.savefig() # 保存
plt. close(0) # 关闭图 0

还有简便的 plt.close('all') 关闭所有图
不用管 fig 号码