python写文件格式转换程序_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)