将子图形保存在.imshow()之后

Question

以下是我的可视化代码：

f, ax = plt.subplots(1, 2)
for i, img in enumerate([img1, img2]):    
grads = # my visualization codes
# visualize grads as heatmap
ax[i].imshow(grads, cmap='jet')

如何保存此处使用imshow显示的内容？非常感谢任何建议！

Answer 1

保存整个图形很简单，只需使用savefig函数：

f.savefig('filename.png')

有许多文件格式可以保存，这些格式通常是从文件名的扩展名中正确推断出来的。有关详细信息，请参阅文档。

savefig函数采用参数bbox_inches，该参数定义要保存的地物区域。要将单个子地块保存到文件中，可以使用子地块的Axes对象的边界框来计算适当的值。

把所有这些放在一起，你的代码看起来像这样：

f, ax = plt.subplots(1, 2)
for i, img in enumerate([img1, img2]):    
grads = # my visualization codes
# visualize grads as heatmap
ax[i].imshow(grads, cmap='jet')
# Save the subplot.
bbox = ax[i].get_tightbbox(f.canvas.get_renderer())
f.savefig("subplot{}.png".format(i),
bbox_inches=bbox.transformed(f.dpi_scale_trans.inverted()))
# Save the whole figure.
f.savefig("whole_figure.png")