Python 中的 numpy.save()

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

语法

参数

file: str, file, 或 pathlib.path

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

allow_pickle : bool（可选）

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

fix_imports : bool（可选）

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

示例 1

输出

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' 和 filename。
• 我们使用 seek(0) 函数关闭并重新打开了文件。
• 最后，我们尝试加载 out_file。

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

示例 2

输出

array([[ 0,  1,  2,  3,  4],
       	[ 5,  6,  7,  8,  9],
      	[10, 11, 12, 13, 14]])