修改这段代码,使数据能存储到数据库中而不是csv文件中
时间: 2024-02-15 20:05:23 浏览: 32
要将数据存储到数据库中,你需要使用相应的数据库API,并创建一个连接到数据库的连接对象。这里我以MySQL为例,你可以根据自己的数据库类型进行修改。
```python
import mysql.connector
# 建立数据库连接
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 创建游标对象
mycursor = mydb.cursor()
# 执行SQL语句,创建数据表
mycursor.execute("CREATE TABLE customers (name VARCHAR(255), address VARCHAR(255))")
# 插入数据
sql = "INSERT INTO customers (name, address) VALUES (%s, %s)"
val = ("John", "Highway 21")
mycursor.execute(sql, val)
# 提交事务
mydb.commit()
# 输出插入的数据行数
print(mycursor.rowcount, "record inserted.")
```
在这个例子中,我们首先使用`mysql.connector`库连接到MySQL数据库。然后,我们创建一个游标对象,用于执行SQL语句。接下来,我们执行了一个SQL语句,创建了一个名为`customers`的数据表。然后,我们插入了一行数据,使用`%s`作为占位符来代替实际的值。最后,我们提交了事务,并输出插入的数据行数。
你可以根据自己的需要修改SQL语句和数据表结构来适应你的数据存储需求。