Python 中變量的保存和讀取

1. 保存為matlab可讀的.mat文件

from scipy.io import savemat
savemat('ab.mat', {'a': a, 'b': b}) 

 2. 使用numpy保存為.npy或者.npz文件

import numpy as np

np.save('a', a = a)
data = np.load('a.npy')
a = data['a']

np.savez('a', a=a, b=b)
data = np.load('a.npz')
a = data['a']
b = data['b']

 3. 使用numpy保存字典

import numpy as np

a = [1, 2, 3]
b = [4, 5, 6]

ab = {'a': a, 'b': b}
np.save('data.npy', ab)

data = np.load('data.npy', allow_pickle=True)
a = data.item()['a'] 
b = data.item()['b']