NumPy 文件数据读写

写数据

NumPy 数组可以使用 np.save 方法保存到本地磁盘中,默认扩展名是 .npy,并且是未压缩的二进制格式。

import numpy as np

a = np.array([[0, 1, 2], [3, 4, 5]])
np.save('test1.npy', a)

如果没有指定文件扩展名,则默认将会是 .npy

如果要使用未压缩的 .npz 格式同时保存多个数组到一个文件中,则可以使用 np.savez 方法:

a = np.arange(4)
b = np.arange(7)

np.savez('test2.npz', arr0=a, arr1=b)

相应的,如果从 .npz 文件中加载数据,将会得到一个类字典对象:

>>> dic  = np.load('test2.npz')
>>> dic['arr0']
array([0, 1, 2, 3])

另一种保存数据到文件中的方法是 np.savetxt,它允许我们设置被保存数据的格式:

>>> x = np.arange(4)
>>> # 这里用逗号作为元素之间的分隔符
>>> np.savetxt('test3.out', x, delimiter=',')

读数据

与写数据对应,从文件中读取数据有两个常用方法:np.loadnp.loadtxt

>>> np.load('test1.npy')
array([[0, 1, 2], [3, 4, 5]])
>>> np.loadtxt('test3.out', delimiter=',')
array([0., 1., 2., 3.])

np.savetxt 类似,np.loadtxt 方法也有一些加载时的选项,具体可以查看文档

你可能感兴趣的:(NumPy 文件数据读写)