对于读限制python中的行

Question

我只想执行此循环rownum次：

rownum = 1000
with open('outputfile.csv', 'wb') as out:
    with open('inputfile.csv', 'rbU') as ifile:
        out_writer = csv.writer(out)
        for row in read:
            out_writer.writerow(row)

插入像for i in range(0, rownum):这样的东西不起作用，但我也无法理解它放在哪里。我怎么说for y in x N times？

Answer 1

使用itertools.islice

for row in itertools.islice(read, stop = rownum):
    out_writer.writerow(row)

Answer 2

如果您不想导入itertools，还可以使用enumerate和break：

    for rowidx, row in enumerate(read):
        if rowidx == MAX_ROWS: break
        out_writer.writerow(row)

itertools更像是pythonic。