在Python中插入数据到数据库通常涉及以下步骤:
导入数据库模块:
根据所使用的数据库类型,导入相应的Python库,如`pymysql`用于MySQL,`psycopg2`用于PostgreSQL,`sqlite3`用于SQLite等。
建立数据库连接:
使用库提供的连接函数,传入数据库地址、用户名、密码等信息来建立连接。
创建游标对象:
通过连接对象的`cursor()`方法创建一个游标对象,用于执行SQL语句。
执行SQL语句:
使用游标对象的`execute()`方法执行插入数据的SQL语句。
提交事务:
如果是插入操作,使用连接对象的`commit()`方法提交事务。
关闭游标和连接:
操作完成后,使用游标对象的`close()`方法关闭游标,使用连接对象的`close()`方法关闭数据库连接。
```python
import pymysql
建立数据库连接
conn = pymysql.connect(
host='localhost',
user='root',
password='',
db='test1',
port=3306
)
创建游标对象
cursor = conn.cursor()
插入单条数据
sql_insert = "INSERT INTO student (id, name, age) VALUES (%s, %s, %s)"
data = (, 'tom', 12)
cursor.execute(sql_insert, data)
提交事务
conn.commit()
关闭游标和连接
cursor.close()
conn.close()
对于批量插入,可以使用库提供的批量插入方法,例如在SQLite中可以使用以下方式:
```python
import sqlite3
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
批量插入数据
data = [
('Alice', 25),
('Bob', 30),
('Charlie', 35)
]
使用executemany方法进行批量插入
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", data)
提交事务
conn.commit()
关闭游标和连接
cursor.close()
conn.close()
请根据你的具体需求选择合适的方法和库进行数据库操作。