Python numpy save用法及代码示例

本文简要介绍 python 语言中 numpy.save 的用法。

用法:

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

将数组保存为 NumPy .npy 格式的二进制文件。

参数：

file： 文件、str 或 pathlib.Path

保存数据的文件或文件名。如果文件是file-object，则文件名不变。如果文件是字符串或路径，如果文件名还没有扩展名.npy，则会将其附加到文件名中。

arr： array_like

要保存的数组数据。

allow_pickle： 布尔型，可选

允许使用 Python pickles 保存对象数组。不允许 pickle 的原因包括安全性(加载 pickle 数据可以执行任意代码)和可移植性(pickled 对象可能无法在不同的 Python 安装上加载，例如，如果存储的对象需要不可用的库，并且并非所有 pickle 数据在Python 2 和 Python 3)。默认值：真

fix_imports： 布尔型，可选

仅在强制 Python 3 上的对象数组中的对象以 Python 2 兼容的方式 pickle 时有用。如果 fix_imports 为 True，pickle 将尝试将新的 Python 3 名称映射到 Python 2 中使用的旧模块名称，以便 Python 2 可以读取 pickle 数据流。

注意：

有关 .npy 格式的说明，请参阅 numpy.lib.format 。

保存到文件中的所有数据都附加到文件末尾。

例子：

>>> from tempfile import TemporaryFile
>>> outfile = TemporaryFile()

>>> x = np.arange(10)
>>> np.save(outfile, x)

>>> _ = 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])

>>> with open('test.npy', 'wb') as f:
...     np.save(f, np.array([1, 2]))
...     np.save(f, np.array([1, 3]))
>>> with open('test.npy', 'rb') as f:
...     a = np.load(f)
...     b = np.load(f)
>>> print(a, b)
# [1 2] [1 3]