lingram2020
diff --git a/Collapse file
‎Work/fileparse.py‎
Copy file name to clipboard
+58Lines changed: 58 additions & 0 deletions b/Collapse file
‎Work/fileparse.py‎
Copy file name to clipboard
+58Lines changed: 58 additions & 0 deletions
@@ -1,3 +1,61 @@
 # fileparse.py
 #
 # Exercise 3.3
+import csv
+
+def parse_csv(filename, select=None, types=None, has_headers=True, delimiter = ','):
+    '''
+    Parse a CSV file into a list of records
+    '''
+    with open(filename) as f:
+        rows = csv.reader(f, delimiter=delimiter)
+        
+        if has_headers == True: 
+            # Read the file headers
+            headers = next(rows)
+
+            # If a column selector was given, find indices of the specified columns.
+            # Also narrow the set of headers used for resulting dictionaries
+            if select:
+                indices = [headers.index(colname) for colname in select]
+                headers = select
+            else:
+                indices = []
+
+            records = []
+            for row in rows:
+                if not row:    # Skip rows with no data
+                    continue
+                # Filter the row if specific columns were selected
+                if indices:
+                    row = [ row[index] for index in indices ]
+                
+                if types:
+                    row = [func(val) for func, val in zip(types, row) ]
+
+                # Make a dictionary
+                record = dict(zip(headers, row))
+                records.append(record)
+        else: # has_headers == False (no headers)
+            if select:
+                indices = [headers.index(colname) for colname in select]
+            else:
+                indices = []
+            records = []
+            for row in rows:
+                if not row:    # Skip rows with no data
+                    continue
+                # Filter the row if specific columns were selected
+                if indices:
+                    row = [ row[index] for index in indices ]
+                
+                if types:
+                    row = [func(val) for func, val in zip(types, row) ]
+
+                # Make a dictionary
+                record = tuple(row)
+                records.append(record)
+
+    return records
+
+