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 可以读取 pickle 数据流。

范例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.
• 我们还从 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]])