python爬虫数据放入了prettytable中,如何将prettytable转为csv文件,不受列与列之间 | 此符号的影响
时间: 2024-03-12 14:49:56 浏览: 137
网络爬虫+csv文件处理
您可以使用prettytable模块的get_string()方法获取prettytable的字符串表示形式,然后使用csv模块将其写入csv文件。以下是一个示例代码:
```python
import csv
from prettytable import PrettyTable
# 创建一个PrettyTable对象
table = PrettyTable()
table.field_names = ["Name", "Age", "Gender"]
table.add_row(["Alice", 25, "Female"])
table.add_row(["Bob", 30, "Male"])
table.add_row(["Charlie", 20, "Male"])
# 获取prettytable的字符串表示形式
table_string = table.get_string()
# 将字符串转为二维列表
table_list = [row.split("|")[1:-1] for row in table_string.split("\n") if row.strip()]
# 将二维列表写入csv文件
with open("output.csv", "w", newline="") as f:
writer = csv.writer(f)
writer.writerows(table_list)
```
在这个例子中,我们创建了一个包含三列的prettytable,并向其中添加了三行数据。然后使用get_string()方法将prettytable转换为字符串,再将字符串转换为二维列表,并将其写入csv文件。注意,在转换为二维列表时,我们使用了字符串的split()方法来去除列与列之间的|符号。最后,我们使用csv.writer()将二维列表写入csv文件。
阅读全文