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

时间:2021-05-22

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

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

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章