使用 NumPy 进行 I/O


ndarray 对象可以保存到磁盘文件中,也可以从磁盘文件中加载。可用的 IO 功能有 -

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

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

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

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 用于在保存到磁盘文件或从磁盘文件读取之前序列化和反序列化对象。

保存txt()

简单文本文件格式的数组数据的存储和检索是通过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() 函数接受其他可选参数,例如页眉、页脚和分隔符。