当前位置: 首页 > 文档资料 > NumPy 中文教程 >

I/O with NumPy

优质
小牛编辑
128浏览
2023-12-01

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

savetxt()

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

例子 (Example)

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()函数接受其他可选参数,如页眉,页脚和分隔符。