详解如何在python中读写和存储matlab的数据文件(*.mat)
背景
在做deeplearning过程中,使用caffe的框架,一般使用matlab来处理图片(matlab处理图片相对简单,高效),用python来生成需要的lmdb文件以及做test产生结果。所以某些matlab从图片处理得到的label信息都会以.mat文件供python读取,同时也python产生的结果信息也需要matlab来做进一步的处理(当然也可以使用txt,不嫌麻烦自己处理结构信息)。
介绍
matlab和python间的数据传输一般是基于matlab的文件格式.mat,python中numpy和scipy提供了一些函数,可以很好的对.mat文件的数据进行读写和处理。
在这里numpy作用是提供Array功能映射matlab里面的Matrix,而scipy提供了两个函数loadmat和savemat来读写.mat文件。
下面是一个简单的测试程序,具体的函数用法可以看帮助文档:
importscipy.ioassio
importmatplotlib.pyplotasplt
importnumpyasnp
#matlab文件名
matfn=u'E:/python/测试程序/162250671_162251656_1244.mat'
data=sio.loadmat(matfn)
plt.close('all')
xi=data['xi']
yi=data['yi']
ui=data['ui']
vi=data['vi']
plt.figure(1)
plt.quiver(xi[::5,::5],yi[::5,::5],ui[::5,::5],vi[::5,::5])
plt.figure(2)
plt.contourf(xi,yi,ui)
plt.show()
sio.savemat('saveddata.mat',{'xi':xi,'yi':yi,'ui':ui,'vi':vi})
示例2
importscipy.ioassio
importnumpyasnp
###下面是讲解python怎么读取.mat文件以及怎么处理得到的结果###
load_fn='xxx.mat'
load_data=sio.loadmat(load_fn)
load_matrix=load_data['matrix']#假设文件中存有字符变量是matrix,例如matlab中save(load_fn,'matrix');当然可以保存多个save(load_fn,'matrix_x','matrix_y',...);
load_matrix_row=load_matrix[0]#取了当时matlab中matrix的第一行,python中数组行排列
###下面是讲解python怎么保存.mat文件供matlab程序使用###
save_fn='xxx.mat'
save_array=np.array([1,2,3,4])
sio.savemat(save_fn,{'array':save_array})#和上面的一样,存在了array变量的第一行
save_array_x=np.array([1,2,3,4])
save_array_y=np.array([5,6,7,8])
sio.savemat(save_fn,{'array_x':save_array_x,'array_x':save_array_x})#同理,只是存入了两个不同的变量供
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。