python arff 转换为mat

python arff 转换为mat
近来见到一种神奇的格式arff格式，需要将这种格式转化为matlab可以读取的数据，即mat文件。
首先我们看一下arff数据是什么样子的：
首先arff数据都有一个索引：

这里记录的数据每一个维度是什么意思。
而下面的数据有两大类：压缩矩阵和非压缩矩阵：

压缩矩阵看起来比较小，因为他只是记录了一部分的信息，就是稀疏矩阵的一部分有信息的数据，没有记录的就默认为0

非压缩的矩阵就比较正常了，因为里面每一条数据的每一个维度都有数据记录。
这里对于非压缩矩阵，我们用以下代码处理：

from scipy.io import arff
import scipy.io as sio
from io import StringIO
import numpy as np
f = open("CAL500.arff")
f = f.read()
f = StringIO(f)
data, meta = arff.loadarff(f)


final = []
for i in range(len(data)):
    now = []
    for j in range(242-174):
        now.append(data[i][j])
    final.append(now)
final=np.array(final)
final=final.astype(float)
sio.savemat('CAL500x.mat', {'x': final})  

final = []
for i in range(len(data)):
    now = []
    for j in range((242-174),len(data[0])):
        now.append(data[i][j])
    final.append(now)
final=np.array(final)
final=final.astype(float)
sio.savemat('CAL500y.mat', {'y': final})  

对于压缩矩阵我们用以下代码处理：

import arff
import scipy.io as sio
import numpy as np
import codecs
file_ = codecs.open('delicious.arff','r','UTF-8')
a = arff.load(file_)
c = a['data']
finalx = []
finaly = []
for i in range(len(c)):
    finalx.append(c[i][0:500])
    finaly.append(c[i][500:]) 
finalx=np.array(finalx)
finaly=np.array(finaly)
finalx=finalx.astype(float)
finaly=finaly.astype(float)
sio.savemat('deliciousx.mat', {'x': finalx})  
sio.savemat('deliciousy.mat', {'y': finaly})  

这里用到的包都在Anaconda中包含了，因此安装了Anaconda就基本可以运行这个python代码了

代码链接：https://github.com/anonymouslycn/arfftomat/
————————————————
版权声明：本文为CSDN博主「ANONYMOUSLYCN」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/geng333abc/article/details/80200179