What is the best way to take a data file that contains a header row and read this row into a named tuple so that the data rows can be accessed by header name?
I was attempting something like this:
import csv
from collections import namedtuple
with open('data_file.txt', mode="r") as infile:
reader = csv.reader(infile)
Data = namedtuple("Data", ", ".join(i for i in reader[0]))
next(reader)
for row in reader:
data = Data(*row)
The reader object is not subscriptable, so the above code throws a TypeError
. What is the pythonic way to reader a file header into a namedtuple?
Use:
and omit the line:
Combining this with an iterative version based on martineau's comment below, the example becomes for Python 2
and for Python 3
Please have a look at
csv.DictReader
. Basically, it provides the ability to get the column names from the first row as you're looking for and, after that, lets you access to each column in a row by name using a dictionary.If for some reason you still need to access the rows as a
collections.namedtuple
, it should be easy to transform the dictionaries to named tuples as follows: