Leggere .csv in Python senza eseguire il loop dell'intero file?

Question

L'unico modo in cui ho visto csv.reader di Python utilizzato è in un ciclo for, che passa attraverso l'intero file senza salvare i valori passati delle variabili read in. Ho solo bisogno di lavorare con 2 file consecutive del (enorme) file alla volta. Usando il csv.reader per loop, ho solo 1 linea alla volta.

C'è un modo per usare il modulo csv di Python per prendere solo una riga di un file csv senza dover finire di leggere il file fino alla fine?

Devo impostare le variabili sui valori nella prima riga, impostare un secondo set di variabili sui valori della riga successiva, utilizzare simultaneamente i due insiemi di variabili per i calcoli, quindi sovrascrivere il primo insieme di variabili con il secondo set, e leggere una nuova riga per sovrascrivere il secondo set.

Answer 1

Non c'è niente che ti costringa a utilizzare il lettore in un ciclo. Basta leggere la prima riga, quindi leggere la seconda riga.

import csv 
rdr = csv.reader(open("data.csv")) 
line1 = rdr.next() # in Python 2, or next(rdr) in Python 3 
line2 = rdr.next() 

Answer 2

La risposta ovvia sembra essere quella di memorizzare la riga precedente su ogni iterazione.

>>> for x in csv.DictReader(stream): 
... print prevLine 
... print x 
... prevLine = x 
.... 

Answer 3

Blatant rubare da TK ... ... per lo più la domanda che rimane è, cosa vuole fare l'OP con la prima e l'ultima riga del file?

prevLine = None 

for x in csv.DictReader(stream): 
    if prevLine is not None: 
     DoWork(prevLine, x) 
    else: 
     Initialize(x) 
    prevLine = x 

Finalize(prevLine) 

Answer 4

Se stai sempre guardando esattamente due righe consecutive, suona per me come si potrebbe trarre vantaggio dall'utilizzo della coppie recipe. Dal modulo itertools:

from itertools import tee, izip 
def pairwise(iterable): 
    "s -> (s0,s1), (s1,s2), (s2, s3), ..." 
    a, b = tee(iterable) 
    next(b, None) 
    return izip(a, b) 

si può usare questa in questo modo:

for first_dict, second_dict in pairwise(csv.DictReader(stream)): 
    # do stuff with first_dict and second_dict 

Answer 5

Leggi CSV:

readCSV = csv.reader(csvFile, delimiter=',') 

Leggere la riga successiva in Python 2.7:

row = readCSV.next() 

Leggi la riga successiva in Python 3.4:

row = readCSV.__next__()