Python+Selenium基础篇之17 - 获取的数据导出excel表格

这章我们介绍一下自动化获取到的数据如何导出excel表格

相关代码如下：

# coding=utf-8
import xlwt
import time
from selenium import webdriver

driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
time.sleep(1)
workbook = xlwt.Workbook(encoding='utf-8')  #创建workbook对象
worksheet = workbook.add_sheet('sheet1')  #创建工作表
worksheet.write(0,0,'id')	#创建工作表头

count = 1
for link in driver.find_elements_by_xpath('//*[@id]'):
    link = link.get_attribute('id') 
    
    worksheet.write(count, 0, link)	# 插入数据到表格
    count = count + 1
workbook.save('student.xls') # excel表格保存

driver.quit()