数据存储到本地或者数据库中
时间: 2023-11-14 19:06:09 浏览: 98
数据本地存储
将爬取的数据存储到本地或者数据库中是一个很重要的步骤,方便后续数据分析和可视化。以下是一个简单的示例代码,演示如何将数据存储到本地文件中:
```python
import csv
# 假设数据已经被存储到了列表 data 中
data = [['景点1', 100, 4.5], ['景点2', 200, 4.3], ['景点3', 300, 4.9]]
# 将数据写入 CSV 文件中
with open('data.csv', 'w', newline='', encoding='utf-8-sig') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['景点名称', '门票价格', '评分'])
for row in data:
writer.writerow(row)
```
以上代码将数据以 CSV 格式保存到了本地文件 data.csv 中。
如果您希望将数据存储到数据库中,可以使用 Python 的数据库操作库,如 MySQLdb、PyMySQL、MongoDB 等库。以下是一个示例代码,演示如何使用 MySQLdb 库将数据存储到 MySQL 数据库中:
```python
import MySQLdb
# 假设数据已经被存储到了列表 data 中
data = [['景点1', 100, 4.5], ['景点2', 200, 4.3], ['景点3', 300, 4.9]]
# 连接到 MySQL 数据库
db = MySQLdb.connect("localhost", "root", "password", "testdb")
# 使用 cursor() 方法获取操作游标
cursor = db.cursor()
# 执行 SQL 语句,创建数据表
cursor.execute("CREATE TABLE IF NOT EXISTS scenery (name VARCHAR(255), price INT, rating FLOAT)")
# 将数据插入到数据表中
for row in data:
name, price, rating = row
sql = "INSERT INTO scenery (name, price, rating) VALUES ('%s', '%d', '%f')" % (name, price, rating)
cursor.execute(sql)
db.commit()
# 关闭数据库连接
db.close()
```
以上代码将数据以表格的形式存储到 MySQL 数据库中,方便后续的数据分析和可视化。
阅读全文