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]