GeezerVison is showing github.com — python_reference/useful_scripts/read_csv_to_dict.py at master · datahacking/python_reference

Expand file tree

read_csv_to_dict.py

Copy path

More file actions

21 lines (18 loc) · 697 Bytes

Copy path

21 lines (18 loc) · 697 Bytes

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

# Sebastian Raschka, 03/2014

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}

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