上一主题

numpy.load

下一主题

numpy.savez

numpy.save

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

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

参数:

文件:文件或str

保存数据的文件或文件名。如果file是一个文件对象,则文件名不变。如果file是一个字符串,则.npy扩展名将附加到文件名后面,如果它还没有。

allow_pickle:bool,可选

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

fix_imports:bool,可选

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

arr:array_like

要保存的数组数据。

也可以看看

savez
将多个数组保存到.npz存档中

savetxtload

笔记

有关.npy格式的说明,请参阅numpy.lib.format的模块docstring或Numpy增强提议http://docs.scipy.org /doc/numpy/neps/npy-format.html

例子

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