NumPy 输入/输出
ndarray对象可以保存到磁盘文件中,并从磁盘文件中加载。可用的IO函数如下:
- load() 和 save() 函数用于处理NumPy二进制文件(扩展名为.npy)
-
loadtxt() 和 savetxt() 函数用于处理普通文本文件
NumPy引入了一种简单的ndarray对象文件格式。该 .npy 文件存储了在磁盘文件中重构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。