jax.numpy.save#

jax.numpy.save(file, arr, allow_pickle=True, fix_imports=<no value>)[原始碼]#

以 NumPy .npy 格式將陣列儲存到二進位檔案。

參數:

  • file (file, str, 或 pathlib.Path) – 要將資料儲存到的檔案或檔案名稱。如果 file 是檔案物件，則檔案名稱保持不變。如果 file 是字串或 Path，且檔案名稱尚未具有 .npy 副檔名，則會附加該副檔名。

  • arr (array_like) – 要儲存的陣列資料。

  • allow_pickle (bool, 選用) – 允許使用 Python pickles 儲存物件陣列。不允許使用 pickles 的原因包括安全性（載入 pickled 資料可能會執行任意程式碼）和可攜性（pickled 物件可能無法在不同的 Python 安裝上載入，例如，如果儲存的物件需要無法使用的程式庫，並且並非所有 pickled 資料都與不同版本的 Python 相容）。預設值：True

  • fix_imports (bool, 選用) –

    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]