numpy.save#

numpy.save(file, arr, allow_pickle=True, fix_imports=<no value>)[source]#

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

参数:
file文件、字符串或 pathlib.Path

数据保存到的文件或文件名。如果 `file` 是文件对象,则文件名不变。如果 `file` 是字符串或 Path 对象,且文件名没有 `.npy` 扩展名,则会自动添加。

arr类数组

要保存的数组数据。

allow_pickle布尔型,可选

允许使用 Python pickle 保存对象数组。不允许使用 pickle 的原因包括安全性(加载 pickle 数据可能会执行任意代码)和可移植性(pickle 对象可能无法在不同的 Python 安装上加载,例如如果存储的对象需要不可用的库,并且并非所有 pickle 数据都与不同版本的 Python 兼容)。默认值:True

fix_imports布尔型,可选

fix_imports 标志已弃用,并且没有作用。

自 2.1 版本弃用:自 NumPy 1.17 以来,此标志已被忽略,它仅用于支持在 Python 2 中加载一些由 Python 3 写入的文件。

另请参阅

savez

将多个数组保存到 .npz 归档文件中

savetxt, load

注意

有关 .npy 格式的描述,请参阅 numpy.lib.format

保存到文件的任何数据都将附加到文件末尾。

示例

>>> import numpy as np
>>> from tempfile import TemporaryFile
>>> outfile = TemporaryFile()
>>> x = np.arange(10)
>>> np.save(outfile, x)
>>> _ = outfile.seek(0) # Only needed 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]