Wie durchläuft man bestimmte Reihen von Zeilen mit Python CSV Reader?

Question

Wie durchläuft man einen bestimmten Bereich von Zeilen mit Python CSV Reader?

Der folgende Code durchläuft alle Zeilen:

with open(trainFile, 'rt') as csvfile: 
    spamreader = csv.reader(csvfile, delimiter=' ', quotechar='|') 
    for row in spamreader: 
     print (', '.join(row)) 

I Schleife wünschen nur aus (I bis J).

Answer 1

können Sie itertools.islice verwenden:

import itertools 

i, j = 10, 20 
with open(trainFile, 'rt') as csvfile: 
    spamreader = csv.reader(csvfile, delimiter=' ', quotechar='|') 
    for row in itertools.islice(spamreader, i, j+1): 
     print (', '.join(row)) 

Alternative (Code folgende ist möglich, weil csv.reader eine iterable annehmen):

HINWEIS : funktioniert nur, wenn CSV-Zeilen keine Zeilenumbrüche enthalten.

import itertools 

i, j = 10, 20 
with open(trainFile, 'rt') as csvfile: 
    spamreader = csv.reader(itertools.islice(csvfile, i, j+1), 
          delimiter=' ', quotechar='|') 
    for row in spamreader: 
     print (', '.join(row)) 

Answer 2

Verwendung islice, zB:

rows_1_to_50 = itertools.islice(spamreader, 0, 50) 
for row in rows_1_to_50: 
    pass 

Answer 3

Eine weitere itertools Implementierung dropwhile und takewhile

from itertools import takewhile, dropwhile 
trainFile = 'x.1' 
low_lim = 3 
high_lim = 6 
with open(trainFile, 'rt') as csvfile: 
    spamreader = csv.reader(csvfile, delimiter=' ', quotechar='|') 
    g = (x for x in enumerate(spamreader, 1)) 
    g = dropwhile(lambda x: x[0] < low_lim, g) 
    g = takewhile(lambda x: x[0] <= high_lim, g) 
    for row in g: 
     print (', '.join(row[1]))