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

Python  /  管理员 发布于 7年前   313

本文实例为大家分享了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) 

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。