python读写.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文件。

示例


  
  
  1. import scipy.io as sio
  2. import numpy as np
  3. ###下面是讲解python怎么读取.mat文件以及怎么处理得到的结果###
  4. load_fn = 'xxx.mat'
  5. load_data = sio.loadmat(load_fn)
  6. load_matrix = load_data[ 'matrix'] #假设文件中存有字符变量是matrix,例如matlab中save(load_fn, 'matrix');当然可以保存多个save(load_fn, 'matrix_x', 'matrix_y', ...);
  7. load_matrix_row = load_matrix[ 0] #取了当时matlab中matrix的第一行,python中数组行排列
  8. ###下面是讲解python怎么保存.mat文件供matlab程序使用###
  9. save_fn = 'xxx.mat'
  10. save_array = np.array([ 1, 2, 3, 4])
  11. sio.savemat(save_fn, { 'array': save_array}) #和上面的一样,存在了array变量的第一行
  12. save_array_x = np.array([ 1, 2, 3, 4])
  13. save_array_y = np.array([ 5, 6, 7, 8])
  14. sio.savemat(save_fn, { 'array_x': save_array_x, 'array_x': save_array_x}) #同理,只是存入了两个不同的变量供使用

猜你喜欢

转载自blog.csdn.net/zseqsc_asd/article/details/84979802