在Python中进行数据库操作通常涉及以下几个步骤:安装数据库适配器(也称为数据库驱动或接口)、连接到数据库、执行SQL语句、处理结果以及关闭连接。下面我将以几个流行的数据库为例,介绍如何使用Python进行数据库操作。
1. SQLite
SQLite是一种轻量级的数据库,它不需要独立的服务器进程或操作,可以直接在Python中使用。Python标准库中的sqlite3
模块提供了对SQLite数据库的支持。
python复制代码
import sqlite3 | |
# 连接到SQLite数据库 | |
# 如果文件不存在,会自动在当前目录创建 | |
conn = sqlite3.connect('example.db') | |
# 创建一个Cursor对象,并通过它执行SQL命令 | |
c = conn.cursor() | |
# 创建表 | |
c.execute('''CREATE TABLE IF NOT EXISTS stocks | |
(date text, trans text, symbol text, qty real, price real)''') | |
# 插入数据 | |
c.execute("INSERT INTO stocks VALUES ('2023-01-01','BUY','RHAT',100,35.14)") | |
# 提交事务 | |
conn.commit() | |
# 查询数据 | |
c.execute('SELECT * FROM stocks WHERE symbol=?', ('RHAT',)) | |
print(c.fetchall()) | |
# 关闭连接 | |
conn.close() |
2. MySQL
对于MySQL数据库,Python提供了mysql-connector-python
和PyMySQL
等第三方库来支持。这里以mysql-connector-python
为例。
首先,你需要安装mysql-connector-python
:
bash复制代码
pip install mysql-connector-python |
然后,你可以像这样连接到MySQL数据库并执行操作:
python复制代码
import mysql.connector | |
# 连接到MySQL数据库 | |
conn = mysql.connector.connect( | |
host="localhost", | |
user="yourusername", | |
password="yourpassword", | |
database="mydatabase" | |
) | |
c = conn.cursor() | |
# 创建表 | |
c.execute("CREATE TABLE IF NOT EXISTS employees (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), position VARCHAR(255))") | |
# 插入数据 | |
c.execute("INSERT INTO employees (name, position) VALUES (%s, %s)", ("John Doe", "Developer")) | |
# 提交事务 | |
conn.commit() | |
# 查询数据 | |
c.execute("SELECT * FROM employees") | |
print(c.fetchall()) | |
# 关闭连接 | |
conn.close() |
3. PostgreSQL
对于PostgreSQL数据库,可以使用psycopg2
库。
首先,安装psycopg2
:
bash复制代码
pip install psycopg2-binary |
然后,连接到PostgreSQL数据库并执行操作:
python复制代码
import psycopg2 | |
# 连接到PostgreSQL数据库 | |
conn = psycopg2.connect( | |
dbname="mydatabase", | |
user="myuser", | |
password="mypassword", | |
host="127.0.0.1", | |
port="5432" | |
) | |
cur = conn.cursor() | |
# 创建表 | |
cur.execute("CREATE TABLE IF NOT EXISTS test (id SERIAL PRIMARY KEY, num INTEGER, data VARCHAR(50))") | |
# 插入数据 | |
cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)", (100, "abc'def")) | |
# 提交事务 | |
conn.commit() | |
# 查询数据 | |
cur.execute("SELECT * FROM test") | |
rows = cur.fetchall() | |
for row in rows: | |
print(row) | |
# 关闭连接 | |
cur.close() | |
conn.close() |
以上是在Python中操作SQLite、MySQL和PostgreSQL数据库的基本步骤。不同的数据库系统可能需要安装不同的库,并且连接字符串(用于连接到数据库的参数)也可能会有所不同。但是,一旦你建立了连接,执行SQL语句和处理结果的基本模式通常是相似的。