### 使用Python连接和查询数据库的方法及示例代码 使用Python连接和查询数据库是开发数据驱动软件的重要技能。以下是几种常见的方法，包括使用`sqlite3`、`pymysql`和`SQLAlchemy`等库的详细说明和示例代码。 #### 1. 使用 `sqlite3` 连接 SQLite 数据库 SQLite 是一种轻量级的嵌入式数据库，适合小型项目或测试环境。Python 内置了 `sqlite3` 模块，可以直接使用它来创建和操作 SQLite 数据库。 ```python import sqlite3 # 创建或连接到一个SQLite数据库文件 conn = sqlite3.connect('example.db') # 创建游标对象 cursor = conn.cursor() # 创建表 cursor.execute(''' CREATE TABLE IF NOT EXISTS users ( id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL ) ''') # 插入数据 cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25)) # 查询数据 cursor.execute("SELECT * FROM users") rows = cursor.fetchall() # 输出查询结果 for row in rows: print(row) # 提交事务并关闭连接 conn.commit() conn.close() ``` 这种方法适用于简单的数据库操作[^2]。 #### 2. 使用 `pymysql` 连接 MySQL 数据库 对于需要连接 MySQL 数据库的场景，可以使用 `pymysql` 库。以下是一个完整的连接和查询示例。 ```python import pymysql def connect_to_mysql(host, user, password, database): # 建立与MySQL数据库的连接 conn = pymysql.connect(host=host, user=user, password=password, database=database) return conn def execute_sql_query(conn, sql_query): # 创建游标对象 cursor = conn.cursor() # 执行SQL查询 cursor.execute(sql_query) # 获取查询结果 result = cursor.fetchall() # 关闭游标 cursor.close() return result if __name__ == '__main__': # 连接到数据库 db_conn = connect_to_mysql('localhost', 'root', 'password', 'mydatabase') # 执行查询 result = execute_sql_query(db_conn, 'SELECT * FROM users') # 打印查询结果 print(result) # 关闭数据库连接 db_conn.close() ``` 此方法展示了如何通过 `pymysql` 连接到 MySQL 数据库并执行查询[^4]。 #### 3. 使用 `SQLAlchemy` 进行 ORM 操作 `SQLAlchemy` 是一个功能强大的 ORM（对象关系映射）工具，能够简化数据库交互。以下是一个基于 SQLAlchemy 的示例。 ```python from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker # 创建引擎并连接到数据库 engine = create_engine('sqlite:///example.db', echo=True) # 定义基础类 Base = declarative_base() # 定义用户模型 class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String) age = Column(Integer) # 创建表 Base.metadata.create_all(engine) # 创建会话 Session = sessionmaker(bind=engine) session = Session() # 添加新用户 new_user = User(name='Bob', age=30) session.add(new_user) session.commit() # 查询用户 users = session.query(User).all() for user in users: print(f'{user.name} is {user.age} years old.') # 关闭会话 session.close() ``` 这种方法利用 ORM 技术将数据库表映射为 Python 类，从而实现更优雅的数据库操作。 #### 4. 使用 `pyodbc` 连接 SQL Server 数据库 对于需要连接 SQL Server 数据库的场景，可以使用 `pyodbc` 库。以下是一个示例。 ```python import pyodbc # 定义连接字符串 conn_str = 'DRIVER={SQL Server};SERVER=服务器名称;DATABASE=数据库名称;UID=用户名;PWD=密码' # 建立连接 conn = pyodbc.connect(conn_str) # 创建游标 cursor = conn.cursor() # 执行查询 cursor.execute('SELECT * FROM 表名') # 获取所有记录 rows = cursor.fetchall() # 遍历结果集并输出每行数据 for row in rows: print(row) # 关闭游标和连接 cursor.close() conn.close() ``` 此方法适用于需要连接 SQL Server 数据库的场景[^5]。 ---