NumPy 的 I/O 操作



ndarray 物件可以儲存到磁碟檔案並從中載入。 可用的 I/O 函式包括:

  • load()save() 函式處理 NumPy 二進位制檔案(副檔名為 npy

  • loadtxt()savetxt() 函式處理普通文字檔案

NumPy 引入了一種簡單的 ndarray 物件檔案格式。此 .npy 檔案將資料、形狀、dtype 和重建 ndarray 所需的其他資訊儲存在磁碟檔案中,以便即使檔案位於具有不同架構的其他機器上也能正確檢索陣列。

numpy.save()

numpy.save() 函式將輸入陣列儲存到副檔名為 npy 的磁碟檔案中。

import numpy as np 
a = np.array([1,2,3,4,5]) 
np.save('outfile',a)

要從 outfile.npy 重建陣列,請使用 load() 函式。

import numpy as np 
b = np.load('outfile.npy') 
print b 

這將產生以下輸出:

array([1, 2, 3, 4, 5])

save() 和 load() 函式接受一個額外的布林引數 allow_pickles。Python 中的 pickle 用於在儲存到或讀取自磁碟檔案之前序列化和反序列化物件。

savetxt()

使用 savetxt()loadtxt() 函式以簡單的文字檔案格式儲存和檢索陣列資料。

示例

import numpy as np 

a = np.array([1,2,3,4,5]) 
np.savetxt('out.txt',a) 
b = np.loadtxt('out.txt') 
print b 

這將產生以下輸出:

[ 1.  2.  3.  4.  5.] 

savetxt() 和 loadtxt() 函式接受其他可選引數,例如 header、footer 和 delimiter。

廣告