利用numpy.save函數保存Python中的數組數據

一、numpy.save函數的介紹

import numpy as np

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

# 使用numpy.save函數保存數組到文件
np.save('a.npy', a)

# 加載保存的數組
b = np.load('a.npy')
print(b)

numpy.save函數是numpy庫中提供的一個函數，用來將一個數組保存到文件中，在需要的時候，我們可以從文件中讀取數組數據。該函數的調用方式非常簡單。其中第一個參數表示保存到的文件名，第二個參數表示要保存的數組。

二、使用numpy.save保存不同形狀的數組

import numpy as np

# 保存一維數組
a = np.array([1,2,3])
np.save('a.npy', a)

# 保存二維數組
b = np.array([[1,2,3],[4,5,6]])
np.save('b.npy', b)

# 保存三維數組
c = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
np.save('c.npy', c)

# 加載保存的數組
a_load = np.load('a.npy')
b_load = np.load('b.npy')
c_load = np.load('c.npy')

print(a_load)
print(b_load)
print(c_load)

numpy.save函數可以保存不同形狀的數組，比如一維數組、二維數組、三維數組等等。不同形狀的數組可以通過給numpy.save傳遞不同的數組參數來保存。

三、使用numpy.save保存不同類型的數組

import numpy as np

# 保存整型數組
a = np.array([1,2,3])
np.save('a.npy', a)

# 保存浮點型數組
b = np.array([[1.0,2.0,3.0],[4.0,5.0,6.0]])
np.save('b.npy', b)

# 保存複數數組
c = np.array([[1+2j,3+4j],[5+6j,7+8j]])
np.save('c.npy', c)

# 加載保存的數組
a_load = np.load('a.npy')
b_load = np.load('b.npy')
c_load = np.load('c.npy')

print(a_load)
print(b_load)
print(c_load)

numpy.save函數同樣可以保存不同類型的數組。整型、浮點型、複數等類型的數組都可以被保存到文件中。在加載這些數組時需要注意它們的數據類型。

四、使用numpy.savez保存多個數組

import numpy as np

# 保存多個數組
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
np.savez('ab.npz', a=a, b=b)

# 加載保存的多個數組
data = np.load('ab.npz')
print(data['a'])
print(data['b'])

numpy.savez函數可以保存多個數組到同一個文件中，保存時需要將每個數組命名，加載時也需要指定相應的數組名進行獲取。