NumPy 文件数据读写

写数据

NumPy 数组可以使用 np.save 方法保存到本地磁盘中，默认扩展名是 .npy，并且是未压缩的二进制格式。

import numpy as np

a = np.array([[0, 1, 2], [3, 4, 5]])
np.save('test1.npy', a)

如果没有指定文件扩展名，则默认将会是 .npy

如果要使用未压缩的 .npz 格式同时保存多个数组到一个文件中，则可以使用 np.savez 方法：

a = np.arange(4)
b = np.arange(7)

np.savez('test2.npz', arr0=a, arr1=b)

相应的，如果从 .npz 文件中加载数据，将会得到一个类字典对象：

>>> dic  = np.load('test2.npz')
>>> dic['arr0']
array([0, 1, 2, 3])

---

另一种保存数据到文件中的方法是 np.savetxt，它允许我们设置被保存数据的格式：

>>> x = np.arange(4)
>>> # 这里用逗号作为元素之间的分隔符
>>> np.savetxt('test3.out', x, delimiter=',')

读数据

与写数据对应，从文件中读取数据有两个常用方法：np.load 和 np.loadtxt：

>>> np.load('test1.npy')
array([[0, 1, 2], [3, 4, 5]])
>>> np.loadtxt('test3.out', delimiter=',')
array([0., 1., 2., 3.])

与 np.savetxt 类似，np.loadtxt 方法也有一些加载时的选项，具体可以查看文档