selenium保存数据
时间: 2023-11-12 16:04:11 浏览: 128
selenium本身并不具备保存数据的功能,但可以通过配合其他库来实现数据的保存。以下是两种常见的保存数据的方法:
1. 写入xlsx文件:可以使用openpyxl库来将数据写入Excel文件中。首先需要安装openpyxl库,然后在代码中导入该库并创建一个工作簿对象,接着将数据写入到工作簿中,最后保存工作簿即可。
2. 写入csv文件:可以使用csv库来将数据写入CSV文件中。首先需要导入csv库并创建一个CSV文件对象,然后将数据写入到CSV文件中,最后关闭文件即可。
示例代码如下:
1. 写入xlsx文件:
```
import openpyxl
# 创建工作簿对象
wb = openpyxl.Workbook()
# 获取工作表对象
ws = wb.active
# 写入数据
ws['A1'] = '姓名'
ws['B1'] = '年龄'
ws['C1'] = '性别'
ws['A2'] = '张三'
ws['B2'] = 18
ws['C2'] = '男'
# 保存工作簿
wb.save('data.xlsx')
```
2. 写入csv文件:
```
import csv
# 创建CSV文件对象
with open('data.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
# 写入表头
writer.writerow(['姓名', '年龄', '性别'])
# 写入数据
writer.writerow(['张三', 18, '男'])
```
阅读全文