Làm thế nào để lặp qua hàng loạt các hàng cụ thể bằng trình đọc csv của Python?

Question

Làm thế nào để lặp qua một loạt các hàng cụ thể với trình đọc csv Python?

Các mã sau vòng qua tất cả các hàng:

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

Tôi muốn để lặp chỉ từ nhất định (i tới j).

Answer 1

Bạn có thể sử dụng itertools.islice:

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 (mã sau đây có thể vì csv.reader chấp nhận một iterable):

LƯU Ý : chỉ hoạt động khi các hàng CSV không chứa dòng mới.

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

Sử dụng islice, ví dụ:

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

Answer 3

Một thực hiện itertools sử dụng dropwhile và 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]))