1. 程式人生 > >python3 讀取檔案跳過檔案第一行內容

python3 讀取檔案跳過檔案第一行內容

Python程式設計時,經常需要跳過第一行讀取檔案內容。比較容易想到是為每行設定一個line_num,然後判斷line_num是否為1,如果不等於1,則進行讀取操作。相應的Python程式碼如下:

#input_file = open("C:\\Python34\\test.csv") #這種寫法讀取完了要進行close操作
with open("C:\\Python34\\test.csv") as input_file:
line_num = 0  
for line in islice(input_file, 1, None):  #設定預設input_file檔案第一行為空
    line_num += 1
if (line_num != 1): do_readline()

但這樣寫的程式碼執行效率偏低,因為每次都需要判斷當前的行號是否為1。使用Python提供的itertools工具,我們可以避免此類問題。itertools的目的就是為了提搞looping的效率。相應的程式碼如下:

from itertools import islice #每次讀取判斷是否第一行會很慢使用此模組會更優
#vip = open("vip.txt") #這種寫法讀取完了要進行close操作
with open("vip.txt") as vip:
    for line
in islice(vip,1,None): #設定讓其預設第一行為空 driver.find_element_by_css_selector("#memberID").send_keys(line.split(',')[0]) driver.find_element_by_css_selector("#abbr").send_keys(line.split(',')[1]) driver.find_element_by_css_selector("#name").send_keys(line.split(',')[2]) driver.find_element_by_css_selector("#eAbbr"
).send_keys(line.split(',')[3]) driver.find_element_by_css_selector("#eName").send_keys(line.split(',')[4])

參考連結