Python+Selenium學習筆記15 - 讀取txt和csv文件

 讀取txt的內容並用百度查找搜索

# coding = utf-8

from selenium import webdriver
import time

# 打開瀏覽器
dr = webdriver.Firefox()
dr.get("http://www.baidu.com")

'''
# 讀取列表的
words = ['python', 'selenium', 'webdriver']

for word in words:
    dr.find_element_by_id("kw").clear()
    dr.find_element_by_id("kw").send_keys(word)
    dr.find_element_by_id("kw").submit()
    time.sleep(5)

'''

# 打開文件並讀取
f = open("word.txt", "r")
words = f.readlines()
f.close()

# 循環遍歷txt內容再逐個查詢
for word in words:
    dr.find_element_by_id("kw").clear()
    dr.find_element_by_id("kw").send_keys(word)
    dr.find_element_by_id("kw").submit()
    time.sleep(5)
    print word


dr.quit()

 txt文件內容

 

讀取CSV文件

import csv

# 讀取本地CSV文件
f = csv.reader(open("file.csv", "r"))

for word in f:
    print word

# 輸出結果
['apple', 'banana', 'cat']
['dulian', 'ege', 'find']
['go', 'hello', 'ip']
['just', 'kind', 'len']
['man', 'noon', '']

CSV文件內容

 

 只讀取某一列

import csv

# 讀取本地CSV文件
f = csv.reader(open("file.csv", "r"))

for word in f:
    print word[1]


# 輸出結果
banana
ege
hello
kind
noon