如何从CSV文件中读取n行

Question

我试图从一个大型CSV文件中读取一些数据的子集，我已经尝试使用nrows，但它返回以下错误消息，'TypeError: 'nrows'是此函数的无效关键字参数'不确定如何纠正它。

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)