matplotlib畫圖保存

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

 

fig = plt.figure(0) # 新圖 0 plt.savefig() # 保存 plt. close(0) # 關閉圖 0 

還有簡便的 plt.close('all') 關閉所有圖
不用管 fig 號碼