small bugfix

rasbt · rasbt · commit 015c6cf0c3c0 · 2014-03-19T14:59:12.000-04:00
diff --git a/read_csv_to_dict.py b/read_csv_to_dict.py
@@ -9,6 +9,9 @@ def read_csv(csv_path):
         data = {i:[] for i in header}
         for line in in_csv:
             line = line.strip().split(',')
-            for i in range(len(line)-1):
-                data[header[i]].append(line[i])
+            for i in range(len(line)):
+                try: 
+                    data[header[i]].append(float(line[i]))
+                except ValueError:
+                    data[header[i]].append(line[i])
     return data
