将python ndarray转换为matlab文件

时间:2013-11-05 20:33:34

标签: python arrays matlab numpy scipy

所以我有一个CSV文件,我已将其读入列表中。我已将该列表转换为数组,并使用以下函数将数组保存到MATLAB文件中。

def save_array(arr,filename):
    import scipy.io
    out_dict={}
    out_dict[filename]=arr
    scipy.io.savemat(filename + '.mat',out_dict)

然而,当我打开MATLAB文件时,出现了问题。当我在Python中打开时,我得到以下输出:

{'M': array([[u'153  ', u'81   ', u'0.28 ', ..., u'0.19 ', u'-0.07', u'1    '],
   [u'168  ', u'76   ', u'0.08 ', ..., u'0.98 ', u'0.42 ', u'0    '],
   [u'184  ', u'92   ', u'0.18 ', ..., u'0.92 ', u'0.75 ', u'0    '],
   ..., 
   [u'183  ', u'62   ', u'0.57 ', ..., u'0.87 ', u'0.31 ', u'0    '],
   [u'181  ', u'72   ', u'0.48 ', ..., u'0.91 ', u'1.2  ', u'0    '],
   [u'158  ', u'77   ', u'1.01 ', ..., u'0.99 ', u'0.88 ', u'0    ']], 
  dtype='<U5'),
 '__globals__': [],
 '__header__': 'MATLAB 5.0 MAT-file Platform: posix, Created on: Tue Nov  5 15:28:57 2013',
 '__version__': '1.0'}

为什么每个元素的开头都有u?我怎么能纠正这个?

1 个答案:

答案 0 :(得分:3)

我看到您正在读取CSV文件并获取字符串数组。 您可以在保存它们之前将它们转换为浮点数数组:

import numpy as np
out_dict[filename]=np.array(arr, dtype=np.float64)