NumPy 的 I/O 操作



ndarray 对象可以保存到磁盘文件并从中加载。 可用的 I/O 函数包括:

  • 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() 函数接受其他可选参数,例如 header、footer 和 delimiter。

广告