Skip to content

Commit 3d62be0

Browse files
author
kleinjohann
committed
Allow for multiple columns
1 parent 54de295 commit 3d62be0

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

neo/rawio/phyrawio.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -233,15 +233,19 @@ def _parse_tsv_or_csv_to_list_of_dict(filename):
233233

234234
for row in reader:
235235
if line == 0:
236-
key1, key2 = tuple(row.keys())
236+
cluster_id_key, *annotation_keys = tuple(row.keys())
237237
# Convert cluster ID to int
238-
row[key1] = int(row[key1])
238+
row[cluster_id_key] = int(row[cluster_id_key])
239239
# Convert strings without letters
240-
if letter_pattern.match(row[key2]) is None:
241-
if float_pattern.match(row[key2]) is None:
242-
row[key2] = int(row[key2])
243-
else:
244-
row[key2] = float(row[key2])
240+
for key in annotation_keys:
241+
value = row[key]
242+
if not len(value):
243+
row[key] = None
244+
elif letter_pattern.match(value) is None:
245+
if float_pattern.match(value) is None:
246+
row[key] = int(value)
247+
else:
248+
row[key] = float(value)
245249

246250
list_of_dict.append(row)
247251
line += 1

0 commit comments

Comments
 (0)