Qual è il modo pietonico per leggere i dati del file CSV come file di namedtuples?

Question

Qual è il modo migliore per prendere un file di dati che contiene una riga di intestazione e leggere questa riga in una tupla denominata in modo che le righe di dati siano accessibili dal nome dell'intestazione?

stavo cercando qualcosa di simile:

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) 

Il lettore oggetto non è subscriptable, in modo che il codice di cui sopra tiri un TypeError. Qual è il modo pietonico per leggere l'intestazione di un file in un namedtuple?

Answer 1

Usa:

Data = namedtuple("Data", next(reader)) 

e omettere la linea:

next(reader) 

Combinando questo con una versione iterativa basata sul commento di Martineau di seguito, l'esempio diventa per Python 2

import csv 
from collections import namedtuple 
from itertools import imap 

with open("data_file.txt", mode="rb") as infile: 
    reader = csv.reader(infile) 
    Data = namedtuple("Data", next(reader)) # get names from column headers 
    for data in imap(Data._make, reader): 
     print data.foo 
     # ...further processing of a line... 

e per Python 3

import csv 
from collections import namedtuple 

with open("data_file.txt", newline="") as infile: 
    reader = csv.reader(infile) 
    Data = namedtuple("Data", next(reader)) # get names from column headers 
    for data in map(Data._make, reader): 
     print(data.foo) 
     # ...further processing of a line... 

Answer 2

Si prega di dare un'occhiata a csv.DictReader. Fondamentalmente, fornisce la possibilità di ottenere i nomi delle colonne dalla prima riga che stai cercando e, successivamente, ti consente di accedere a ciascuna colonna di una riga per nome usando un dizionario.

Se per qualche ragione è ancora necessario per accedere alle righe come un collections.namedtuple, dovrebbe essere facile per trasformare i dizionari di tuple denominati come segue:

with open('data_file.txt') as infile: 
    reader = csv.DictReader(infile) 
    Data = collections.namedtuple('Data', reader.fieldnames) 
    tuples = [Data(**row) for row in reader]