Python面试题:在 Python 中,如何连接并操作数据库?

在 Python 中,可以使用多种库来连接并操作数据库。常用的库有 sqlite3psycopg2(用于 PostgreSQL)、PyMySQL(用于 MySQL)、SQLAlchemy(一个 ORM 工具,支持多种数据库)。下面以 SQLite 和 MySQL 为例,介绍如何连接并操作数据库。

使用 sqlite3 库操作 SQLite 数据库

  1. 连接数据库
python 复制代码
import sqlite3

# 连接到 SQLite 数据库(如果数据库不存在,会自动创建)
conn = sqlite3.connect('example.db')

# 创建游标对象
cursor = conn.cursor()
  1. 创建表
python 复制代码
# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users
                  (id INTEGER PRIMARY KEY AUTOINCREMENT,
                   name TEXT NOT NULL,
                   age INTEGER NOT NULL)''')
  1. 插入数据
python 复制代码
# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")

# 提交事务
conn.commit()
  1. 查询数据
python 复制代码
# 查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()

for row in rows:
    print(row)
  1. 更新数据
python 复制代码
# 更新数据
cursor.execute("UPDATE users SET age = 26 WHERE name = 'Bob'")

# 提交事务
conn.commit()
  1. 删除数据
python 复制代码
# 删除数据
cursor.execute("DELETE FROM users WHERE name = 'Alice'")

# 提交事务
conn.commit()
  1. 关闭连接
python 复制代码
# 关闭游标和连接
cursor.close()
conn.close()

使用 PyMySQL 库操作 MySQL 数据库

  1. 安装 PyMySQL
sh 复制代码
pip install pymysql
  1. 连接数据库
python 复制代码
import pymysql

# 连接到 MySQL 数据库
conn = pymysql.connect(
    host='localhost',
    user='yourusername',
    password='yourpassword',
    database='yourdatabase'
)

# 创建游标对象
cursor = conn.cursor()
  1. 创建表
python 复制代码
# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users
                  (id INT AUTO_INCREMENT PRIMARY KEY,
                   name VARCHAR(255) NOT NULL,
                   age INT NOT NULL)''')
  1. 插入数据
python 复制代码
# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")

# 提交事务
conn.commit()
  1. 查询数据
python 复制代码
# 查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()

for row in rows:
    print(row)
  1. 更新数据
python 复制代码
# 更新数据
cursor.execute("UPDATE users SET age = 26 WHERE name = 'Bob'")

# 提交事务
conn.commit()
  1. 删除数据
python 复制代码
# 删除数据
cursor.execute("DELETE FROM users WHERE name = 'Alice'")

# 提交事务
conn.commit()
  1. 关闭连接
python 复制代码
# 关闭游标和连接
cursor.close()
conn.close()

使用 SQLAlchemy 库操作数据库

  1. 安装 SQLAlchemy
sh 复制代码
pip install sqlalchemy
  1. 连接数据库
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, autoincrement=True)
    name = Column(String)
    age = Column(Integer)

# 创建数据表
Base.metadata.create_all(engine)

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()
  1. 插入数据
python 复制代码
# 插入数据
new_user = User(name='Alice', age=30)
session.add(new_user)
session.commit()
  1. 查询数据
python 复制代码
# 查询数据
users = session.query(User).all()
for user in users:
    print(user.name, user.age)
  1. 更新数据
python 复制代码
# 更新数据
user = session.query(User).filter_by(name='Alice').first()
user.age = 31
session.commit()
  1. 删除数据
python 复制代码
# 删除数据
user = session.query(User).filter_by(name='Alice').first()
session.delete(user)
session.commit()

这三个示例展示了如何使用 sqlite3PyMySQLSQLAlchemy 库连接并操作数据库。根据具体需求和数据库类型,可以选择合适的库来实现数据库操作。

相关推荐
羱滒3 分钟前
sql调优之数据库开发规范
java·数据库·数据库开发
程序员Android6 分钟前
MTK多帧拍照流程分析
数据库
不剪发的Tony老师7 分钟前
Apache Doris:一款高性能的实时数据仓库
数据库·数据仓库
甜可儿19 分钟前
redis序列化设置
数据库·redis
GGBondlctrl23 分钟前
【SpringBoot】论坛项目中如何进行实现发布文章,以及更新对应数据库的数据更新
数据库·springboot项目·三层设计思想·文章发布项目
若云止水37 分钟前
Ubuntu 下 nginx-1.24.0 源码分析 - ngx_init_cycle 函数 - 详解(4)
数据库·nginx·ubuntu
紫雾凌寒38 分钟前
计算机视觉 |解锁视频理解三剑客——TimeSformer
python·深度学习·神经网络·计算机视觉·transformer·timesformer
jay丿1 小时前
Django模板系统深入
数据库·django·sqlite
程序员杰哥1 小时前
测试用例详解
自动化测试·软件测试·python·功能测试·测试工具·职场和发展·测试用例
千层冷面2 小时前
Redis除了做缓存还能做什么?
数据库·redis·缓存