numpy中保存加载数据的方式

保存/加载npy

全局函数形式：

numpy.save("filename.npy",a)
a = numpy.load("filename")

保存/加载txt

保存/加载npz

如果想多组数组保存到一个文件中，可以使用np.savez，其第一个参数为文件名，其后面是需要保存的数组，也可以是使用关键字参数为数组起一个名字，非关键字参数传递的数组会自动起名为arr_0, arr_1；

savez函数输出的是一个压缩文件（扩展名为.npz），其中每个文件都是一个save函数保存的.npy文件，文件名对应于数组名；

load函数自动识别.npz文件，并返回一个类似于字典的对象，可以通过数组名作为关键字获取数组的内容；

import numpy as np

if __name__ == '__main__':
    ######################################################
    ## 保存数据 ##
    a = np.arange(3)
    b = np.arange(4)
    c = np.arange(5)
    np.savez('array_save.npz', a, b, c_array=c)

    ## 加载数据 ##
    A = np.load('array_save.npz')
    # print(type(A))  # <class 'numpy.lib.npyio.NpzFile'>

    print(A['arr_0'])
    print(A['arr_1'])
    print(A['c_array'])

保存/加载pickle

pickle能保存数据, 类型不改变

np.load("some.pkl", allow_pickle=True)