Diff of /dataprocess/csvTools.py [000000] .. [f77492]

Switch to side-by-side view

--- a
+++ b/dataprocess/csvTools.py
@@ -0,0 +1,47 @@
+'''
+provided by LUNA16
+evaluation script
+'''
+
+import csv
+import platform
+
+def writeTXT(filename, lines):
+    with open(filename, 'w') as f:
+        for line in lines:
+            f.write(line+'\n')
+
+def writeCSV(filename, lines):
+    with open(filename, "w", newline='') as f:
+        csvwriter = csv.writer(f)
+        csvwriter.writerows(lines)
+
+def readCSV(filename):
+    lines = []
+    with open(filename, "r") as f:
+        csvreader = csv.reader(f)
+        for line in csvreader:
+            lines.append(line)
+    return lines
+
+def tryFloat(value):
+    try:
+        value = float(value)
+    except:
+        value = value
+    
+    return value
+
+def getColumn(lines, columnid, elementType=''):
+    column = []
+    for line in lines:
+        try:
+            value = line[columnid]
+        except:
+            continue
+            
+        if elementType == 'float':
+            value = tryFloat(value)
+
+        column.append(value)
+    return column