知行编程网知行编程网  2022-09-15 13:00 知行编程网 隐藏边栏  1,045 
文章评分 0 次,平均分 0.0
导语: 本文主要介绍了关于python 获取csv文件的前几行的相关知识,包括python 写入csv,以及python读取csv文件的某一列这些编程知识,希望对大家有参考作用。

python获取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指定读取方式。编写代码如下:

#!/usr/bin/python 
import pandasas pd

data = pd.read_csv('data.csv',nrows =5)
print(data)

代码的运行结果如下:

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$

从上面的结果可以看出,预期的功能是通过指定要读取的行数来实现的。

本文为原创文章,版权归所有,欢迎分享本文,转载请保留出处!

知行编程网
知行编程网 关注:1    粉丝:1
这个人很懒,什么都没写
扫一扫二维码分享