NumPy 文件數據讀寫

寫數據

NumPy 數組可以使用 np.save 方法保存到本地磁盤中，默認擴展名是 .npy，並且是未壓縮的二進制格式。

import numpy as np

a = np.array([[0, 1, 2], [3, 4, 5]])
np.save('test1.npy', a)

如果沒有指定文件擴展名，則默認將會是 .npy

如果要使用未壓縮的 .npz 格式同時保存多個數組到一個文件中，則可以使用 np.savez 方法：

a = np.arange(4)
b = np.arange(7)

np.savez('test2.npz', arr0=a, arr1=b)

相應的，如果從 .npz 文件中加載數據，將會得到一個類字典對象：

>>> dic  = np.load('test2.npz')
>>> dic['arr0']
array([0, 1, 2, 3])

---

另一種保存數據到文件中的方法是 np.savetxt，它允許我們設置被保存數據的格式：

>>> x = np.arange(4)
>>> # 這里用逗號作為元素之間的分隔符
>>> np.savetxt('test3.out', x, delimiter=',')

讀數據

與寫數據對應，從文件中讀取數據有兩個常用方法：np.load 和 np.loadtxt：

>>> np.load('test1.npy')
array([[0, 1, 2], [3, 4, 5]])
>>> np.loadtxt('test3.out', delimiter=',')
array([0., 1., 2., 3.])

與 np.savetxt 類似，np.loadtxt 方法也有一些加載時的選項，具體可以查看文檔