在 Google Colaboratory 上保存或下载 matplotlib 绘图图像

Posted 2023-03-16

【中文标题】在 Google Colaboratory 上保存或下载 matplotlib 绘图图像

【英文标题】：Saving or downloading matplotlib plot images on Google Colaboratory

【发布时间】：2018-11-29 22:04:33

【问题描述】：

我在 Google Colaboratory 中使用 matplotlib 并尝试将绘图图像保存在某处（在本地下载或上传到 Google Drive）。我目前正在内联显示图像：

plt.show()

这是我尝试过的，虽然它只下载一个空白图像：

import os    
local_download_path = os.path.expanduser('~/data')
plot_filepath = os.path.join(local_download_path, "plot.png")

plt.savefig(plot_filepath)

from google.colab import files
files.download(plot_filepath)

我也尝试使用 Drive API 上传绘图图像，但也没有成功。

【问题讨论】：

可以直接右键保存图片吗？

是的，但我正在生成许多图。

【参考方案1】：

在使用savefig() 之前，您确定没有显示情节吗？这是一个完整的工作示例：

import matplotlib.pyplot as plt
from google.colab import files

test = plt.figure()
plt.plot([[1, 2, 3], [5, 2, 3]])
plt.savefig('test.pdf')

files.download('test.pdf')

【讨论】：

我遇到了同样的问题，这解决了。我 plt.show() 在保存它之前保存它在我自己的计算机上工作但在 colab 上没有。

@lhhong 我也一样

【参考方案2】：

您需要显示绘图并指定要保存的绘图。否则，您只是在保存一个空白的绘图画布。

import matplotlib.pyplot as plt
from google.colab import files

test = plt.figure()
plt.plot([[1, 2, 3], [5, 2, 3]])
plt.figure(figsize=(50,50))
test.show()

test.savefig('samplefigure.png')
files.download('samplefigure.png')

【参考方案3】：

如果您想要高分辨率图像，请使用此代码：

  import matplotlib.pyplot as plt
  from google.colab import files

  image = plt.figure(figsize=(16,10), dpi= 200)
  image.savefig('myimage.png',  bbox_inches="tight")
  files.download('myimage.png')