用selenium爬取数据后导入csv文件
时间: 2023-01-12 13:13:05 浏览: 223
csv 文件导入导出
4星 · 用户满意度95%
在使用 Selenium 爬取数据后,可以使用 Python 的 csv 模块将数据导入 CSV 文件。
首先,需要在代码中导入 csv 模块:
```
import csv
```
然后,可以使用 csv 模块的 `writer` 函数创建一个 `csv.writer` 对象,并使用它的 `writerows` 方法将数据写入 CSV 文件:
```python
# 将数据写入 CSV 文件
with open('data.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(data)
```
其中,`data` 是一个包含数据的列表,每一行数据应该是一个列表,列表中的每一个元素对应一个单元格。
例如,如果要将下面的表格写入 CSV 文件:
| 姓名 | 年龄 | 性别 |
|------|------|------|
| 张三 | 20 | 男 |
| 李四 | 22 | 女 |
则可以使用如下代码:
```python
data = [
['姓名', '年龄', '性别'],
['张三', 20, '男'],
['李四', 22, '女']
]
with open('data.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(data)
```
这样就可以将数据写入 CSV 文件了。
阅读全文