Numpy保存和导入文件

文件读写

Numpy可以方便的进行文件读写，如下面这种格式的文本文件：

Numpy保存和导入文件 - 图1

  1. # 使用np.fromfile从文本文件'housing.data'读入数据
  2. # 这里要设置参数sep = ' '，表示使用空白字符来分隔数据
  3. # 空格或者回车都属于空白字符，读入的数据被转化成1维数组
  4. d = np.fromfile('./work/housing.data', sep = ' ')
  5. d
  1. array([6.320e-03, 1.800e+01, 2.310e+00, ..., 3.969e+02, 7.880e+00,
  2.        1.190e+01])

文件保存

Numpy提供了save和load接口，直接将数组保存成文件(保存为.npy格式)，或者从.npy文件中读取数组。

  1. # 产生随机数组a
  2. a = np.random.rand(3,3)
  3. np.save('a.npy', a)
  5. # 从磁盘文件'a.npy'读入数组
  6. b = np.load('a.npy')
  8. # 检查a和b的数值是否一样
  9. check = (a == b).all()
  10. check
  1. True