zoukankan      html  css  js  c++  java
  • numpy 学习:保存和加载

    numpy使用save(), savez()和load()函数来保存和加载numpy的二进制文件。

    一,保存文件

    save()和savez()的区别是前者是非压缩的,而后者是压缩存储。

    1,save()函数

    def save(file, arr, allow_pickle=True, fix_imports=True)
    • save()函数:以.npy格式将数组保存到二进制文件中。
    • .npy格式:以二进制的方式存储文件,在二进制文件第一行以文本形式保存了数据的元信息(ndim,dtype,shape等),可以用二进制工具查看内容。

    举个例子,把文件保存到二进制文件中。

    outfile = r'.\test.npy'
    np.random.seed(20200619)
    x = np.random.uniform(low=0, high=1,size = [3, 5])
    np.save(outfile, x)
    y = np.load(outfile)
    print(y)

    2,压缩存储

    def savez(file, *args, **kwds):
    • savez()函数:以未压缩的.npz格式将多个数组保存到单个文件中。
    • .npz格式:以压缩打包的方式存储文件,可以用压缩软件解压。
    • savez()函数:第一个参数是文件名,其后的参数都是需要保存的数组,也可以使用关键字参数为数组起一个名字,非关键字参数传递的数组会自动起名为arr_0, arr_1, …
    • savez()函数:输出的是一个压缩文件(扩展名为.npz),其中每个文件都是一个save()保存的.npy文件,文件名对应于数组名。load()自动识别.npz文件,并且返回一个类似于字典的对象,可以通过数组名作为关键字获取数组的内容。

    举个例子:将多个数组保存到一个文件

    outfile = r'.\test.npz'
    x = np.linspace(0, np.pi, 5)
    y = np.sin(x)
    z = np.cos(x)
    np.savez(outfile, x, y, z_d=z)
    data = np.load(outfile)
    np.set_printoptions(suppress=True)
    print(data.files)  
    # ['z_d', 'arr_0', 'arr_1']
    
    print(data['arr_0'])
    # [0.         0.78539816 1.57079633 2.35619449 3.14159265]
    
    print(data['arr_1'])
    # [0.         0.70710678 1.         0.70710678 0.        ]
    
    print(data['z_d'])
    # [ 1.          0.70710678  0.         -0.70710678 -1.        ]

    二,加载文件

    def load(file, mmap_mode=None, allow_pickle=False, fix_imports=True, encoding='ASCII'):
    • load()函数:从.npy.npz或 pickled文件加载数组或pickled对象。
    • mmap_mode: {None, ‘r+’, ‘r’, ‘w+’, ‘c’};:读取文件的方式。
    • allow_pickle=False:允许加载存储在.npy文件中的pickled对象数组。
    • fix_imports=True:若为True,pickle将尝试将旧的python2名称映射到python3中使用的新名称。
    • encoding='ASCII':制定编码格式,默认为“ASCII”。
    作者悦光阴
    本文版权归作者和博客园所有,欢迎转载,但未经作者同意,必须保留此段声明,且在文章页面醒目位置显示原文连接,否则保留追究法律责任的权利。
  • 相关阅读:
    BZOJ2956: 模积和——整除分块
    BZOJ1257: [CQOI2007]余数之和——整除分块
    数位DP【模板】
    2019HDU多校第7场——构造
    AtCoder Grand Contest 032 B
    P3599 Koishi Loves Construction——构造题
    CF C. Vladik and fractions——构造题
    RMQ问题【模板】
    libevent多线程使用事项
    Linux查看进程运行的完整路径方法
  • 原文地址:https://www.cnblogs.com/ljhdo/p/15766989.html
Copyright © 2011-2022 走看看