Function read_csv_as_columns

Solutions/2_6/colreader.py:19–28

(filename, types)

def read_csv_as_columns(filename, types):
    columns = collections.defaultdict(list)
    with open(filename) as f:
        rows = csv.reader(f)
        headers = next(rows)
        for row in rows:
            for name, func, val in zip(headers, types, row):
                columns[name].append(func(val))
            
    return DataCollection(columns)

if __name__ == '__main__':
    import tracemalloc