如何用matplotlib画图并保存图像

1. 首先，import头文件如下：

 

import matplotlib  
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib.pyplot import plot,savefig

2. 画图时：

 

 

fig, ax = plt.subplots(figsize=(12, 12))
ax.imshow(im, aspect='equal') #im是要显示的图像

3. 保存图片：

 

 

savefig("D:/1.jpg")

note:

 

画一张图就执行一次savefig命令，这与命令：

 

plt.show()

不同，这条命令只需执行一次，之前画的所有图就都会展示出来