我有一个包含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,我将如何只读取前1000个条目?
我尝试没有成功:
entries = []
with open('mnist_train.csv', 'r') as f:
mycsv = csv.reader(f)
for row in mycsv[:1000]:
entries.append(row)
如您所见,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)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句