Python pandas读取CSV文件指定前几行方法

月亮慢半拍
月亮慢半拍
发布时间:2026-10-13| 7人浏览过
原创

本文介绍如何利用Python pandas库中的read_csv函数及nrows参数,高效读取大型CSV文件的前几行数据。通过具体代码示例,展示如何指定行数限制,避免加载全部数据,从而优化内存使用和处理速度。

存储数据的CSV文件有时数据量非常庞大,但在实际应用中,我们往往不需要处理全部数据,而只需关注其中的前几行。

借助pandas库中read_csv函数的行数指定功能,可以轻松实现这一需求。

假设存在一个名为data.csv的文件,其内容结构如下所示:

GreydeMac-mini:chapter06 greyzhang$ cat data.csv 
,name_01,coment_01,,,,
2,name_02,coment_02,,,,
3,name_03,coment_03,,,,
4,name_04,coment_04,,,,
5,name_05,coment_05,,,,
6,name_06,coment_06,,,,
7,name_07,coment_07,,,,
8,name_08,coment_08,,,,
9,name_09,coment_09,,,,
10,name_10,coment_10,,,,
11,name_11,coment_11,,,,
12,name_12,coment_12,,,,
13,name_13,coment_13,,,,
14,name_14,coment_14,,,,
15,name_15,coment_15,,,,
16,name_16,coment_16,,,,
17,name_17,coment_17,,,,
18,name_18,coment_18,,,,
19,name_19,coment_19,,,,
20,name_20,coment_20,,,,
21,name_21,coment_21,,,,

若仅需获取前5行数据,可通过设置nrows参数来指定读取范围。具体代码实现如下:

 1 #!/usr/bin/python
 2 
 3 import pandasas pd
 4 
 5 data = pd.read_csv('data.csv',nrows =5)
 6 print(data)
 7 

执行上述代码后,输出结果如下:

GreydeMac-mini:chapter06 greyzhang$ python row_test.py 
 Unnamed: 0 name_01 coment_01 Unnamed: 3 Unnamed: 4 Unnamed: 5 \
0   2 name_02 coment_02   NaN   NaN   NaN 
1   3 name_03 coment_03   NaN   NaN   NaN 
2   4 name_04 coment_04   NaN   NaN   NaN 
3   5 name_05 coment_05   NaN   NaN   NaN 
4   6 name_06 coment_06   NaN   NaN   NaN 
 Unnamed: 6 
0   NaN 
1   NaN 
2   NaN 
3   NaN 
4   NaN 
GreydeMac-mini:chapter06 greyzhang$ 

从运行结果可见,通过指定nrows参数成功实现了仅读取前几行数据的预期功能。

以上内容为本文分享的全部知识,希望能对读者有所帮助,也欢迎大家继续支持本站获取更多技术资源。

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。