2021-05-25

python numpy 格式数据存储

假设 train 为 numpy 数据

import numpy
# 假设numpy数据为train
train = numpy. array([1.0, 1.0, 1.0, 1.0, 2.0], detype=float)

法一：tofile\fromfile

train.tofile("filename.bin")
b = numpy.fromfile("filename.bin", detype=float)
# 存储了一个名字为filename.bin的文件
# 然后用fromfile从文件中读取数据出来

注意：储存的bin文件是二进制文件。因此在文件编解码的时候需要保持数据格式的统一即dtype=需要相同。

法二：save\load

numpy.save("filename.npy", train)
b= numpy.load("filename.npy")
# 将train存储为filename.npy文件
# 然后用load方法读取出来

savez方法可以同时存储多个numpy数据且类型可以不同

train2 = np.array([1, 2, 3, 4])
numpy.savez("filename.npy", train, other=train2)
# 同时存储了两个数据并对train2进行了重命名

load后可以通过名字进行数据的索引
两种方法都可以用load读取

b = numpy.load("filename.npy")
b["train"]
b["other"]

法三：savetxt\loadtxt

numpy.savetxt("filename.txt", train, fmt='%0.8f')
b = numpy.loadtxt("filename.txt", detype=numpy.float32)

用于处理一维和二维数据。