📌  相关文章
📜  Python numpy.save()

📅  最后修改于: 2020-10-27 09:16:32             🧑  作者: Mango

Python的numpy.save()

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

句法：

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

参数：

文件：str，文件或pathlib.path

此参数定义将在其中保存数据的文件或文件名。如果此参数是文件的对象，则文件名将保持不变。如果file参数是路径或字符串，则将.npy扩展名添加到文件名，如果没有扩展名，则将其扩展名。

allow_pickle：bool(可选)

此参数用于允许将对象保存到泡菜中。安全性和可能性是不允许泡菜的原因。

fix_imports：bool(可选)

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

范例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])

在上面的代码中：

  • 我们导入了别名为np的numpy。
  • 我们还从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]])