plt模块结合numpy数组的可视化的常见用法
import matplotlib.pyplot as plt
img = np.array([[1, 2], [3, 4]])
plt.imshow(img)
plt.show()
以上写法在linux或者ubuntu系统中不会有显示,这是因为系统没有图形界面。
解决方法:保存成png图片然后在程序运行后查看,如下:
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
img = np.array([[1, 2], [3, 4]])
plt.imshow(img)
plt.savefig("/home/img_save_folder/Picture.png")
#这里是要保存的路径/home/img_save_folder/和保存文件名Picture.png
如果要同时保存多个图片,为防止覆盖,可以如下添加plt.close()以及为图片名编号:
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
for i in range(5):
img = np.random.randint(10,100,size=(4,3)) #4行3列,每个元素是从10~99之间的随机数)
plt.imshow(img)
plt.savefig("/home/img_save_folder/Picture"+str(i)+".png")
#这里是要保存的路径/home/img_save_folder/和保存文件名Picture0.png、Picture1.png...
plt.close()