Erste Datensatz von CSV-Datei mit DictReader

Question

Ich versuche, den ersten Datensatz als Diktat aus einer CSV-Datei mit dem DictReader zu bekommen. Ich bin nicht in der Lage zu verstehen, wie die Dokumentation nur über Iterieren den Leser Objekt

with open(filename, 'r') as f_in: 
     # Use the csv library to set up a DictReader object. 
     trip_reader = csv.DictReader(f_in) 
     # Use a function on the DictReader object to read the 
     # first trip from the data file and store it in a variable. 
     for row in trip_reader: 
        pprint(row) 

Gibt es eine Funktion erhalten den ersten Datensatz als trip_reader [0] sprechen?

Answer 1

Da Sie über trip_reader laufen kann, können Sie next() auf sie rufen Sie die nächste zu kommen (in diesem Fall die erste Zeile):

with open(filename, 'r') as f_in: 
    # Use the csv library to set up a DictReader object. 
    trip_reader = csv.DictReader(f_in) 
    # Use a function on the DictReader object to read the 
    # first trip from the data file and store it in a variable. 
    row = next(trip_reader) 
    pprint(row) 

Answer 2

Zum ersten Zeile einer CSV ohne for Schleife zu erhalten:

with open(filename, 'r', newline='') as f: 
    r = csv.reader(f, delimiter=',') 
    return next(r) 

Natürlich, wenn Sie eine Reihe von Header haben Sie haben zu „springen“ darueber:

with open(filename, 'r', newline='') as f: 
    r = csv.reader(f, delimiter=',') 
    _ = next(r) # hold header row in throwaway variable 
    return next(r) 

Die newline='' im Kontextmanager wird in Python3 + verwendet (möglicherweise auch spätere Versionen von Python2), aber für Python 2.7.8 und früher können Sie weglassen.