Question

我正在Google合作实验室中使用matplotlib，并尝试将绘图图像保存在某个位置（在本地下载或上传到Google云端硬盘）。我目前正以以下方式内嵌显示图像：

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上传绘图图像，但那里也没有成功。

Answer 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')

Answer 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')

Answer 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')

在Google Colaboratory中保存或下载matplotlib图图像

3 个答案: