NumPy 的 I/O

ndarray 对象可以保存到磁盘文件或从磁盘文件加载。 可用的 IO 函数有 −

  • load()save() 函数处理 numPy 二进制文件(扩展名为 npy

  • loadtxt()savetxt() 函数处理普通文本文件

NumPy 为 ndarray 对象引入了一种简单的文件格式。 此 .npy 文件在磁盘文件中存储数据、形状、dtype 和重建 ndarray 所需的其他信息,这样即使文件位于具有不同体系结构的另一台机器上,也能正确检索数组。


numpy.save()

numpy.save() 文件将输入数组存储在扩展名为 npy 的磁盘文件中。

import numpy as np 
a = np.array([1,2,3,4,5]) 
np.save('outfile',a)

要从 outfile.npy 重建数组,请使用 load() 函数。

import numpy as np 
b = np.load('outfile.npy') 
print b 

它将产生以下输出 −

array([1, 2, 3, 4, 5])

save() 和 load() 函数接受一个额外的布尔参数 allow_pickles。 Python 中的 pickle 用于在保存到磁盘文件或从磁盘文件读取之前序列化和反序列化对象。


savetxt()

以简单文本文件格式存储和检索数组数据是通过 savetxt()loadtxt() 函数完成的。

示例

import numpy as np 

a = np.array([1,2,3,4,5]) 
np.savetxt('out.txt',a) 
b = np.loadtxt('out.txt') 
print b 

它将产生以下输出 −

[ 1.  2.  3.  4.  5.] 

savetxt() 和 loadtxt() 函数接受额外的可选参数,例如页眉、页脚和分隔符。