用于存储数据的csv文件有时候数据量是十分庞大的,然而我们有时候并不需要全部的数据,我们需要的可能仅仅是前面的几行。
这样就可以通过pandas中read_csv中指定行数读取的功能实现。
相关推荐:《Python入门教程》
例如有data.csv文件,文件的内容如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
|
如果我们需要的数据仅仅是前5行,那么读取方式可以通过nrows的方式进行指定。编写代码如下:
1 2 3 4 5 |
|
代码的运行结果如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
|
从上面的结果中可以看出,通过指定读取行数实现了预期的功能。