如何读取csv文件中的前1000个条目

Question

我有一个包含60,000个条目的csv文件.我读了它们并存储在这样的嵌套列表中:

entries = []
with open('mnist_train.csv', 'r') as f:
    mycsv = csv.reader(f)
    for row in mycsv:
        entries.append(row)

而不是阅读所有60,000,我怎么会只阅读前千条？

我尝试了这个没有成功:

entries = []
with open('mnist_train.csv', 'r') as f:
    mycsv = csv.reader(f)
    for row in mycsv[:1000]:
        entries.append(row)

Answer 1

正如您所发现的， acsv.reader不支持切片。您可以使用itertools.islice()通过可迭代的对象来完成此操作。例如，

import itertools

entries = []
with open('mnist_train.csv', 'r') as f:
    mycsv = csv.reader(f)
    for row in itertools.islice(mycsv, 1000):
        entries.append(row)