Update read_csv_to_dict.py

This commit is contained in:
Sebastian Raschka 2014-03-19 20:48:23 -04:00
parent 28692922d4
commit 34d609cc88

View File

@ -1,9 +1,13 @@
# 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):
"""
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, ...], ...}
"""
with open(csv_path, 'r') as in_csv:
header = in_csv.readline().strip().split(',')
data = {i:[] for i in header}