当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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]

相关用法


注:本文由纯净天空筛选整理自numpy.org大神的英文原创作品 numpy.save。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。