python數據如何保存為npy和npz格式并讀取

Python數據如何保存為npy和npz格式并讀取

在Python中，NumPy庫提供了多種方法來保存和加載數組數據。其中，.npy和.npz是兩種常見的文件格式，分別用于保存單個NumPy數組和多個NumPy數組。本文將詳細介紹如何使用這兩種格式保存和讀取數據。

1. 保存和讀取.npy文件

.npy文件格式用于保存單個NumPy數組。使用numpy.save()函數可以將數組保存為.npy文件，而使用numpy.load()函數可以讀取.npy文件。

1.1 保存為.npy文件

import numpy as np

# 創建一個NumPy數組
data = np.array([1, 2, 3, 4, 5])

# 保存數組為.npy文件
np.save('data.npy', data)

上述代碼將數組data保存為data.npy文件。

1.2 讀取.npy文件

import numpy as np

# 讀取.npy文件
loaded_data = np.load('data.npy')

print(loaded_data)

輸出結果為：

[1 2 3 4 5]

2. 保存和讀取.npz文件

.npz文件格式用于保存多個NumPy數組。使用numpy.savez()函數可以將多個數組保存為.npz文件，而使用numpy.load()函數可以讀取.npz文件。

2.1 保存為.npz文件

import numpy as np

# 創建多個NumPy數組
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

# 保存多個數組為.npz文件
np.savez('arrays.npz', array1=array1, array2=array2)

上述代碼將array1和array2保存為arrays.npz文件。

2.2 讀取.npz文件

import numpy as np

# 讀取.npz文件
loaded_arrays = np.load('arrays.npz')

# 訪問保存的數組
print(loaded_arrays['array1'])
print(loaded_arrays['array2'])

輸出結果為：

[1 2 3]
[4 5 6]

3. 總結

• .npy文件格式用于保存單個NumPy數組，使用numpy.save()保存，使用numpy.load()讀取。
• .npz文件格式用于保存多個NumPy數組，使用numpy.savez()保存，使用numpy.load()讀取，并通過鍵名訪問各個數組。

通過這兩種文件格式，可以方便地保存和加載NumPy數組數據，適用于各種數據分析和科學計算場景。