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)

来源：https://blog.csdn.net/chichoxian/article/details/41937581