在Python中,将数据写入数据库通常涉及以下步骤:
1. 导入必要的数据库连接库,例如 `sqlite3`、`pymysql`、`psycopg2` 等。
2. 建立与数据库的连接。
3. 创建数据库表(如果表不存在)。
4. 准备要插入的数据。
5. 执行插入数据的SQL语句,可以使用参数化查询以提高安全性。
6. 提交事务(如果有必要)。
7. 关闭数据库连接。
import pymysql
连接数据库
connection = pymysql.connect(
host='localhost',
user='username',
password='password',
database='database_name',
charset='utf8mb4' 防止中文乱码
)
获取游标
cursor = connection.cursor()
准备要插入的数据
data = [
('John', 25, ''),
('Alice', 30, ''),
('Bob', 35, '')
]
SQL插入语句
sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
执行批量插入
cursor.executemany(sql, data)
提交事务
connection.commit()
关闭游标和连接
cursor.close()
connection.close()
对于其他数据库,如SQLite或PostgreSQL,步骤类似,只需替换相应的连接库和连接参数即可。例如,使用 `sqlite3`:
import sqlite3
连接数据库
conn = sqlite3.connect('example.db')
创建游标
cursor = conn.cursor()
创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
name TEXT,
age INTEGER
)
''')
插入数据
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))
提交更改
conn.commit()
关闭连接
conn.close()