将numpy矩阵写入文件

Question

我正在尝试将numpy数组写入文件。但它不起作用。我的问题如下。

我正在尝试使用matplotlib.image获取图片的RGB像素值，并将从我的程序获得的值保存到某个文件中以备将来使用。

这些值以数组的形式返回，我尝试使用以下代码将其写入文本文件。但该文件无法打开。它也没有显示错误信息。

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

img=mpimg.imread('kitten5.jpeg')
print(img)

f=open('kitten.dat','w')
f.write(img)
f.close()

imgplot=plt.imshow(img)
plt.show()

有人可以帮忙吗？并非必须将数据单独保存到文本文件中。其他建议同样受欢迎。

Answer 1

要将图像保存为numpy二进制文件，您可能需要使用：

np.save('outputfile', img)

关于np.save的好处是你可以在以后使用：

恢复你的numpy对象

np.load('outputfile.npy')

你并不需要关心形状，类型等。