diff --git a/Code_reticent b/Code_reticent deleted file mode 100644 index b6f7f31..0000000 --- a/Code_reticent +++ /dev/null @@ -1,32 +0,0 @@ -import csv -import Snt2Int - - -# 目前通过文件读取,可作为头文件调用 - -##################config################### -trainDataPath = ["train1.csv","train2.csv"] -########################################### -DataDic = {} - - -def ReadFile(): - for path in trainDataPath: - csvData = csv.reader(open(path)) - for row in list(csvData)[1:]: - tmp = row[0].split('_') - Idx,Speaker,Snt = tmp[1],tmp[-1],row[1] - if Idx not in DataDic: - DataDic.update({Idx:[]}) - #DataDic[Idx].append(Speaker+' : '+Snt) - DataDic[Idx].append(Snt) - -def Data2csv(path): - with open(path, 'w', newline='', encoding='utf-8') as f: - writer = csv.writer(f) - writer.writerows([["id","text"]]) - for key in DataDic.keys(): - writer = csv.writer(f) - writer.writerows([[key,"\n".join(DataDic[key]),"\n".join([Snt2Int.Snt2Int(i) for i in DataDic[key]])]]) -ReadFile() -Data2csv('./test_Snt2Int.csv') \ No newline at end of file