python-3.x - 如何读取 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

正如您发现的 csv.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)