3.6

Author: kavun

Work/fileparse.py

@@ -1,33 +1,28 @@
 # fileparse.py
 #
 # Exercise 3.3
-
 import csv
 
 
-def parse_csv(filename, select: list = [], types: list = []):
+def parse_csv(
+    filename: str, select: list = [], types: list = [], has_headers: bool = True
+) -> list:
     """
     Parse a CSV file into a list of records
     """
     with open(filename) as f:
         rows = csv.reader(f)
-
-        # Read the file headers
-        file_headers = next(rows)
-        headers = select if select else file_headers
-        indices = (
-            [headers.index(name) for name in select]
-            if select
-            else [i for i in range(len(headers))]
-        )
+        if has_headers:
+            orig_headers = next(rows)
+            headers = select if select else orig_headers
+            indices = [headers.index(name) for name in headers]
         records = []
-        for row in rows:
-            if not row:  # Skip rows with no data
+        for orig_row in rows:
+            if not orig_row:
                 continue
-            select_row = [row[col] for col in indices]
+            row = [orig_row[i] for i in indices] if has_headers else orig_row
             if types:
-                select_row = [func(col) for func, col in zip(types, select_row)]
-            record = dict(zip(headers, select_row))
+                row = [func(field) for func, field in zip(types, row)]
+            record = dict(zip(headers, row)) if has_headers else tuple(row)
             records.append(record)
-
     return records

Work/fileparse_test.py

@@ -35,3 +35,38 @@ def test_parse_csv_select_3_5():
         {"name": "MSFT", "shares": 50},
         {"name": "IBM", "shares": 100},
     ]
+
+
+def test_parse_csv_no_headers_3_6():
+    assert parse_csv("Data/prices.csv", types=[str, float], has_headers=False) == [
+        ("AA", 9.22),
+        ("AXP", 24.85),
+        ("BA", 44.85),
+        ("BAC", 11.27),
+        ("C", 3.72),
+        ("CAT", 35.46),
+        ("CVX", 66.67),
+        ("DD", 28.47),
+        ("DIS", 24.22),
+        ("GE", 13.48),
+        ("GM", 0.75),
+        ("HD", 23.16),
+        ("HPQ", 34.35),
+        ("IBM", 106.28),
+        ("INTC", 15.72),
+        ("JNJ", 55.16),
+        ("JPM", 36.9),
+        ("KFT", 26.11),
+        ("KO", 49.16),
+        ("MCD", 58.99),
+        ("MMM", 57.1),
+        ("MRK", 27.58),
+        ("MSFT", 20.89),
+        ("PFE", 15.19),
+        ("PG", 51.94),
+        ("T", 24.79),
+        ("UTX", 52.61),
+        ("VZ", 29.26),
+        ("WMT", 49.74),
+        ("XOM", 69.35),
+    ]