Python numpy.save()用法及代碼示例

Python 的 numpy 模塊提供了一個名為 numpy.save() 的函數，可以將數組保存為 .npy 格式的二進製文件。在許多情況下，我們需要二進製格式的數據來操作它。

用法:

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

參數：

文件：str、文件或 pathlib.path

此參數定義將要保存數據的文件或文件名。如果此參數是文件的對象，則文件名將保持不變。如果 file 參數是路徑或字符串，則會在文件名中添加 .npy 擴展名，沒有時會添加。

allow_pickle:bool(可選)

此參數用於允許將對象保存到泡菜中。安全性和概率是不允許泡菜的原因。

fix_imports:bool(可選)

如果 fix_imports 設置為 True，pickle 會將新的 Python3 名稱映射到 Python2 中使用的舊模塊名稱。這使得 Python2 可以讀取 pickle 數據流。

範例1：

import numpy as np
from tempfile import TemporaryFile
out_file = TemporaryFile()
x=np.arange(15)
np.save(out_file, x)
_=out_file.seek(0) # Only needed here to simulate closing & reopening file
np.load(outfile)

輸出：

array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])

在上麵的代碼中：

• 我們已經導入了別名為 np.
• 我們還從 tempfile 導入了 TemporaryFile。
• 我們已經創建了一個 TemporaryFile 的對象 out_file。
• 我們使用 arange() 函數創建了一個數組 'x'。
• 我們使用 np.save() 函數將數組的元素作為二進製保存在 npy 文件中。
• 我們已經在函數中傳遞了數組 'x' 和文件名。
• 我們已使用seek(0) 函數關閉並重新打開文件。
• 最後，我們嘗試加載 out_file。

在輸出中，顯示了一個數組，其中包含 out_file.npy 中存在的元素。

範例2：

import numpy as np
from tempfile import TemporaryFile
outfile = TemporaryFile()
x=np.arange(15)
np.save(outfile, x, allow_pickle=False)
_=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],
      	[10, 11, 12, 13, 14]])