在科學(xué)計(jì)算的過程中，往往需要保存一些數(shù)據(jù)，也經(jīng)常需要把保存的這些數(shù)據(jù)加載到程序中，在 Matlab 中我們可以用 save 和 lood 函數(shù)很方便的實(shí)現(xiàn)。類似的在 Python 中，我們可以用 numpy.save() 和 numpy.load() 函數(shù)達(dá)到類似的效果，并且還可以用 scipy.io.savemat() 將數(shù)據(jù)保存為 .mat 格式，用scipy.io.loadmat() 讀取 .mat 格式的數(shù)據(jù)，達(dá)到可以和 Matlab 或者Octave 進(jìn)行數(shù)據(jù)互動(dòng)的效果.

下面分別介紹之：

numpy.save() 和 numpy.load()numpy.save(arg_1,arg_2) #需要兩個(gè)參數(shù)，arg_1 是文件名，arg_2 是要保存的數(shù)組. 如：import numpy as npa=np.mat('1,2,3;4,5,6')b=np.array([[1,2,3],[4,5,6]])np.save('a.npy',a)np.save('b.npy',b)

這個(gè)時(shí)候 Python 的當(dāng)前工作路徑下就會(huì)多出 a.npy 和 b.npy 兩個(gè)文件，當(dāng)然我們也可以給出具體的路徑，如 np.save(‘D:/PythonWork/a.npy’,a)

下面我們把保存的這兩個(gè)數(shù)據(jù)文件導(dǎo)入到Python :

''' 遇到問題沒人解答？小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群：579817333 尋找有志同道合的小伙伴，互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書！ ''' data_a=np.load('a.npy') data_b=np.load('b.npy') print ('data_a \n',data_a,'\n the type is',type(data_a)) print ('data_b \n',data_a,'\n the type is',type(data_b))data_a [[1 2 3] [4 5 6]] the type is <class 'numpy.ndarray'> data_b [[1 2 3] [4 5 6]] the type is <class 'numpy.ndarray'>

我們可以看到這一過程把原本為矩陣的 a 變?yōu)閿?shù)組型了


如果想同時(shí)保存 a b 到同一個(gè)文件，我們可以用 np.savez() 函數(shù)，具體用法如下：

''' 遇到問題沒人解答？小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群：579817333 尋找有志同道合的小伙伴，互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書！ ''' np.savez('ab.npz',k_a=a,k_b=b)c=np.load('ab.npz')print (c['k_a'])print (c['k_b']) [[1 2 3][4 5 6]][[1 2 3][4 5 6]]

這時(shí)的 c 是一個(gè)字典，需要通過關(guān)鍵字取出我們需要的數(shù)據(jù)

下面我們來認(rèn)識(shí)下 scipy.io.savemat() 和 scipy.io.loadmat()

首先我們用 scipy.io.savemat() 創(chuàng)建 .mat 文件，該函數(shù)有兩個(gè)參數(shù),一個(gè)文件名和一個(gè)包含變量名和取值的字典.

import numpy as npfrom scipy import ioa=np.mat('1,2,3;4,5,6')b=np.array([[1,1,1],[2,2,2]])io.savemat('a.mat', {'matrix': a})io.savemat('b.mat', {'array': b})

總結(jié)

以上是生活随笔為你收集整理的Python中数据的保存和读取的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。