如何从大型CSV文件读取指定行数？解决nrows参数报错问题

错误原因 #

Python标准库的csv.reader()函数不支持nrows参数，这个参数是pandas库中read_csv()的专属参数，直接传入会触发TypeError。

解决方案 #

以下两种方法都能实现“读取前350行并转为列表嵌套列表”的需求：

方法一：使用标准库csv+itertools（无需额外安装）

利用itertools.islice直接截取迭代器的前N行，效率较高，适合处理大文件：

from csv import reader
from itertools import islice

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

如果不想引入itertools，也可以手动循环读取：

from csv import reader

with open('file.csv') as csv_file:
    csv_reader = reader(csv_file)
    lines = []
    for _ in range(350):
        try:
            lines.append(next(csv_reader))
        except StopIteration:
            # 文件行数不足350时自动停止
            break
    print(lines)

方法二：使用pandas库（简洁直观）

如果你的环境允许安装pandas，用read_csv()的nrows参数可以快速实现需求：

import pandas as pd

# 读取前350行
df = pd.read_csv('file.csv', nrows=350)
# 转换为列表嵌套列表格式
lines = df.values.tolist()
print(lines)

内容的提问来源于stack exchange，提问作者ShaqSD