Python技术迷

NumPy教程-numpy.save()在Python中的使用

整理:python架构师

Python的numpy模块提供了一个名为numpy.save()的函数,用于将数组保存到以.npy格式的二进制文件中。在许多情况下,我们需要以二进制格式保存数据以进行操作。

语法:

numpy.save(file, arr, allow_pickle=True, fix_imports=True)

参数:

file: str、file或pathlib.path

此参数定义了要保存数据的文件或文件名。如果此参数是文件对象,则文件名将保持不变。如果file参数是路径或字符串,则会在文件名后添加.npy扩展名,除非它已经有了扩展名。

allow_pickle : bool(可选)

此参数用于允许将对象保存到pickle中。出于安全和概率的原因,不允许使用pickles。

fix_imports : bool(可选)

如果fix_imports设置为True,则pickle会将新的Python3名称映射到Python2中使用的旧模块名称。这使得pickle数据流在Python2中可读。

专属福利 

👉点击领取:最全Python资料合集

示例 1:

import numpy as np  from tempfile import TemporaryFile  out_file = TemporaryFile()  x=np.arange(15)  np.save(out_file, x)  _=out_file.seek(0) # Only needed here to simulate closing & reopening file  np.load(outfile)
输出:
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])
在上面的代码中:
  • 我们导入了numpy,并使用别名np。

  • 我们还从tempfile导入了TemporaryFile。

  • 我们创建了一个TemporaryFile的对象out_file。

  • 我们使用arange()函数创建了一个名为'x'的数组。

  • 我们使用np.save()函数将数组的元素保存为二进制格式在.npy文件中。

  • 我们将数组'x'和文件名传递给函数。

  • 我们使用seek(0)函数关闭并重新打开了文件。

  • 最后,我们尝试加载out_file。

输出显示了一个数组,其中包含在out_file.npy中的元素。

示例 2:

import numpy as np  from tempfile import TemporaryFile  outfile = TemporaryFile()  x=np.arange(15)  np.save(outfile, x, allow_pickle=False)  _=outfile.seek(0) # Only needed here to simulate closing & reopening file  np.load(outfile)
输出:
array([[ 0,  1,  2,  3,  4],       [ 5,  6,  7,  8,  9],       [10, 11, 12, 13, 14]])
Image
 1

Image

热门推荐
Image