如何在txt文件中保存矩阵，然后在python中再次将其作为矩阵打开？

Question

我试过这个：

  import numpy as np
  import os

  outdir= "directory"

  a = np.array([[1,2,3],[1,2,3]])

  os.chdir(outdir)

  np.savetxt("Image.bin", a)

  f = open("directory/Image.bin")
  m = np.fromfile(f, dtype=np.uint16)
  print len(m)
  ma = np.array(np.reshape(m, (2,3)))

  print ma

但它返回此错误消息：“新数组的总大小必须保持不变”

我试图改变dtype，但它不起作用

Answer 1

你应该使用np.loadtxt ： http ： np.loadtxt

尝试：

import numpy as np

a = np.array([[1,2,3],[1,2,3]])
np.savetxt("Image.bin", a)
m = np.loadtxt("Image.bin")

m现在包含在相同的阵列作为a 。

Answer 2

numpy内置了用于保存和加载数组作为二进制文件的函数。

numpy.save('data.npy', data)

将创建该文件（如果你没有，它将附加npy），和

data = numpy.load('data.npy')

将从文件中加载它。 这比将它们保存为文本文件更快，更节省空间。