python数组文件_numpy数组之存取文件的实现示例

将 numpy 数组存入文件，有多种文件类型可供选择，对应地就有不同的方法来读写。

下面我将介绍读写 numpy 的三类文件：

txt 或者 csv 文件

npy 或者 npz 文件

hdf5 文件

通过 numpy 读写 txt 或 csv 文件

import numpy as np

a = np.array(range(20)).reshape((4, 5))

print(a)

# 后缀改为 .txt 一样

filename = 'data/a.csv'

# 写文件

np.savetxt(filename, a, fmt='%d', delimiter=',')

# 读文件

b = np.loadtxt(filename, dtype=np.int32, delimiter=',')

print(b)

缺点：

只能保存一维和二维 numpy 数组，当 numpy 数组 a 有多维时，需要将其 a.reshape((a.shape[0], -1)) 后才能用这种方式保存。

不能追加保存，即每次 np.savetxt() 都会覆盖之前的内容。

通过 numpy 读写 npy 或 npz 文件

读写 npy 文件

import numpy as np

a = np.array(range(20)).reshape((2, 2, 5))

print(a)

filename = 'data/a.npy'

# 写文件<