From ca6688c7358656e1099fab67ade8ac28b3cbc4d3 Mon Sep 17 00:00:00 2001
From: rasbt <se.raschka@me.com>
Date: Wed, 19 Mar 2014 14:54:27 -0400
Subject: [PATCH] csv to dict

---
 read_csv_to_dict.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 read_csv_to_dict.py

diff --git a/read_csv_to_dict.py b/read_csv_to_dict.py
new file mode 100644
index 0000000..9db9b84
--- /dev/null
+++ b/read_csv_to_dict.py
@@ -0,0 +1,14 @@
+# Sebastian Raschka, 03/2014
+# Read a csv file (with a header) into a dictionary
+# where the dictionary structure will be
+#   {header_col1: [val1_line1, val1_line2, ...], header_col2: [val2_line1, val2_line2, ...], ...}
+
+def read_csv(csv_path):
+    with open(csv_path, 'r') as in_csv:
+        header = in_csv.readline().strip().split(',')
+        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[1])
+    return data