python 如何獲取csv文件的前幾行

這篇文章運(yùn)用簡(jiǎn)單易懂的例子給大家介紹python 如何獲取csv文件的前幾行，內(nèi)容非常詳細(xì)，感興趣的小伙伴們可以參考借鑒，希望對(duì)大家能有所幫助。

用于存儲(chǔ)數(shù)據(jù)的csv文件有時(shí)候數(shù)據(jù)量是十分龐大的，然而我們有時(shí)候并不需要全部的數(shù)據(jù)，我們需要的可能僅僅是前面的幾行。

這樣就可以通過(guò)pandas中read_csv中指定行數(shù)讀取的功能實(shí)現(xiàn)。

例如有data.csv文件，文件的內(nèi)容如下：

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,,,,

如果我們需要的數(shù)據(jù)僅僅是前5行，那么讀取方式可以通過(guò)nrows的方式進(jìn)行指定。編寫(xiě)代碼如下：

#!/usr/bin/python 
import pandasas pd

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

代碼的運(yùn)行結(jié)果如下：

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$

從上面的結(jié)果中可以看出，通過(guò)指定讀取行數(shù)實(shí)現(xiàn)了預(yù)期的功能。

關(guān)于python 如何獲取csv文件的前幾行就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò)，可以把它分享出去讓更多的人看到。