数据库是按照数据结构来组织、存储和管理数据的仓库。实际的项目开发中会经常用到数据库。
Python提供了多种库来与各种类型的数据库进行交互,包括关系型数据库(如MySQL、PostgreSQL、SQLite)和非关系型数据库(如MongoDB)。
SQLite
SQLite是一个轻量级的关系型数据库,通常用于嵌入式系统和小型应用。Python内置了对SQLite的支持,通过sqlite3模块即可使用。
import sqlite3
# 连接数据库(不存在则创建)
conn = sqlite3.connect('test.db')
cursor = conn.cursor()
# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
age INTEGER
)
''')
# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ("Alice", 25))
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ("Bob", 30))
# 查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
print(row) # 输出:(1, 'Alice', 25), (2, 'Bob', 30)
# 更新数据
cursor.execute("UPDATE users SET age = ? WHERE name = ?", (26, "Alice"))
# 删除数据
cursor.execute("DELETE FROM users WHERE name = ?", ("Bob",))
# 提交事务并关闭连接
conn.commit()
cursor.close()
conn.close()
MySQL
MySQL是一个非常流行的开源关系型数据库。web网站大量使用。Python通过第三方库(如mysql-connector-python、PyMySQL、mysqlclient)与MySQL进行交互。
安装数据库驱动:使用PyMySQL或mysql-connector-python库:
import pymysql
# 连接数据库
conn = pymysql.connect(
host='localhost',
user='root',
password='your_password',
database='test_db'
)
# 其他的数据库操作方法都一样
PostgreSQL
PostgreSQL是一个功能强大的开源关系型数据库。Python通过psycopg2、asyncpg等库与PostgreSQL进行交互。
pip install psycopg2
import psycopg2
# 连接数据库
conn = psycopg2.connect(
host='localhost',
user='postgres',
password='your_password',
database='test_db'
)
cursor = conn.cursor()
MongoDB
MongoDB是一个流行的非关系型数据库(NoSQL)。Python通过pymongo库与MongoDB进行交互。
pip install pymongo
from pymongo import MongoClient
# 连接数据库
client = MongoClient('mongodb://localhost:27017/')
db = client['test_db']
collection = db['users']
# 插入数据
user1 = {"name": "Alice", "age": 25}
user2 = {"name": "Bob", "age": 30}
collection.insert_many([user1, user2])
# 查询数据
for user in collection.find():
print(user) # 输出:{'_id': ObjectId('...'), 'name': 'Alice', 'age': 25}
# 更新数据
collection.update_one({"name": "Alice"}, {"$set": {"age": 26}})
# 删除数据
collection.delete_one({"name": "Bob"})
# 关闭连接
client.close()