python numpy 写入、读取 .npz 压缩文件

写入npz文件参考文档：https://numpy.org/doc/stable/reference/generated/numpy.savez.html
读取npz文件参考文档：https://numpy.org/doc/stable/reference/generated/numpy.load.html

案例

import numpy as np


def save_zpn():  # 保存.npz文件
    a = np.zeros(10)
    b = np.ones(5)
    c = np.arange(3)
    np.savez('my_npz_file.npz', zeros=a, ones=b, arange=c) # 这里可以指定文件名与保存的文件内容


def load_zpn():  # 读取.npz文件
    my_npz = np.load('my_npz_file.npz')
    print('npz文件保存的文件包括：', my_npz.files) # 打印全部文件名
    print(my_npz['zeros']) # 拿到文件对应的内容
    print(my_npz['ones'])
    print(my_npz['arange'])


if __name__ == '__main__':
    save_zpn()
    load_zpn()