python实现txt文件格式转换为arff格式_Python

python实现txt文件格式转换为arff格式

2021-02-27 00:30君的名字 Python

这篇文章主要为大家详细介绍了python实现txt文件格式转换为arff格式的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

本文实例为大家分享了python实现txt文件格式转换为arff格式的具体代码，供大家参考，具体内容如下

将文件读取出来的时候默认都是字符型的，所以有转换出来有点问题，但是还是可以用的。

文件要求第一行是你对应的属性名，之后是数字。

									import sys 

									import re 

									relationname = "" 

									filename = "" 

									if (len(sys.argv)<2): 

									  print("Usage:\npython arff.py MyRelationName filename.txt") 

									else: 

									  relationname = sys.argv[1] 

									  filename = sys.argv[2] 

									class Arff: 

									  def __init__(self, r, f): 

									    self.relationname = r if r is not "" else "MachineLearning" 

									    f = f if f is not "" else "MMG_data.txt" 

									    self.file1 = open(f, 'r') 

									    self.data = [] 

									    self.names = [] 

									    self.parseData() 

									    self.writeToFile() 

									  def parseData(self): 

									    firstLine = True

									    for line in self.file1.readlines(): 

									      if not firstLine: 

									        try: 

									          line = line.replace("\n", "") 

									          words = line.split(" ") 

									        except ValueError: 

									          print("cant parse file!!") 

									        self.data.append(words) 

									      else: 

									        firstLine = False

									        line = line.replace("\n", "") 

									        words = line.split(" ") 

									        self.names = words 

									  def getType(self, value): 

									    v = "" 

									    if(type(value) == type(1)): 

									      v = "numeric"

									    elif(type(value) == type(1.0)): 

									      v = "numeric"

									    elif(re.match("[0-9]{4}\-[0-9]{2}\-[0-9]{2}\s[0-9]{2}\:[0-9]{2}\:[0-9]{2}", value)): 

									      v = "date " + "yyyy-MM-dd HH:mm:ss"

									    elif(type(value) == type("string")): 

									      v = "string"

									    elif(v == ""): 

									      print("Data type "+value+" not supported yet.") 

									    return v 

									  def writeToFile(self): 

									    values = self.data[0] 

									    file2 = open("Dexhunter_test_result.arff", 'w+' ) 

									    self.relationname+="\n"

									    relationString = '@RELATION ' + self.relationname 

									    file2.write(''+relationString+'') 

									    for i in range(len(self.names)): 

									      str2 = "@ATTRIBUTE " + self.names[i] + " " + self.getType( values[i] ) + "\n"

									      file2.write(''+str2+'') 

									    file2.write('''''@DATA\n''') 

									    for line in self.data: 

									      try: 

									        file2.write(",".join(line)+"\n") 

									      except UnicodeEncodeError: 

									          print("cant write Data to file!!") 

									Arff(relationname, filename)