python3 讀取檔案跳過檔案第一行內容
阿新 • • 發佈:2019-01-03
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])