如何使用python将复杂的.mat文件转换为csv

时间:2017-10-19 04:35:39

标签: python mat-file

我正在尝试使用python将.mat文件转换为csv。我正在使用的代码是

import scipy.io
import numpy as np

data = scipy.io.loadmat("wiki.mat")

for i in data:
    if '__' not in i and 'readme' not in i:
        np.savetxt(("file.csv"),data[i],delimiter=',')

当我运行此代码时,我收到如下错误:

Traceback (most recent call last):
  File "test.py", line 8, in <module>
    np.savetxt(("file.csv"),data[i],delimiter=',')
  File "/Library/Python/2.7/site-packages/numpy/lib/npyio.py", line 1258, in savetxt
    % (str(X.dtype), format))
TypeError: Mismatch between array dtype ('[('dob', 'O'), ('photo_taken', 'O'), ('full_path', 'O'), ('gender', 'O'), ('name', 'O'), ('face_location', 'O'), ('face_score', 'O'), ('second_face_score', 'O')]') and format specifier ('%.18e')

我正在尝试从此链接转换.mat文件:https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/static/imdb_meta.tar

请帮我解决一些有效的解决方案!

2 个答案:

答案 0 :(得分:1)

https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.savetxt.html

  

将数组保存到文本文件中。

不幸的是,您只能在单个文件中存储单个数字numpy数组。而您的.mat文件包含结构:

>> fieldnames(imdb)
                        ans = 
                        {
                          [1,1] = dob
                          [2,1] = photo_taken
                          [3,1] = full_path
                          [4,1] = gender
                          [5,1] = name
                          [6,1] = face_location
                          [7,1] = face_score
                          [8,1] = second_face_score                          
                          [9,1] = celeb_names
                          [10,1] = celeb_id
                        }
>> imdb.name(1)        
                        ans = 
                        {
                          [1,1] = Fred Astaire
                        }

将数据转换为numpy字典(如&#34; Complex matlab-like data structure in python (numpy/scipy)&#34;)中所述,并使用How do I convert this list of dictionaries to a csv file? [Python]将其存储为.csv可能是有意义的

答案 1 :(得分:0)

我创建了一个名为 matgrab 的包,可用于将任何 matlab 数据文件转换为 Dataframe。你可以这样称呼它:

import matgrab
matgrab.mat2df(file.mat).to_csv(file.csv)