mirror of
https://github.com/rasbt/python_reference.git
synced 2025-02-22 00:02:11 +00:00
18 lines
671 B
Python
18 lines
671 B
Python
# 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)):
|
|
try:
|
|
data[header[i]].append(float(line[i]))
|
|
except ValueError:
|
|
data[header[i]].append(line[i])
|
|
return data
|