如何从 CSV 文件中读取 n 行 [重复]

Question

我正在尝试从大型 CSV 文件中读取某些数据的子集，我曾尝试使用 nrows，但它返回以下错误消息，'TypeError: 'nrows' is an invalid keyword argument for this function' 不确定如何更正它。

with open('file.csv') as csv_file:
    csv_reader = reader(csv_file, nrows=350)
    lines = list(csv_reader)
    print(lines) 

#Aim was to print 350 rows in a list of lists

TypeError: 'nrows' is an invalid keyword argument for this function

Answer 1

例如，您可以使用

itertools.islice

从

csv_reader

:

中仅读取 350 行

from csv import reader
from itertools import islice

with open('file.csv', 'r') as csv_file:
    csv_reader = reader(csv_file)
    lines = list(islice(csv_reader, 0, 350))
    print(lines)