Python 操作 SQLite 数据库

SQLite 是一个轻量级的嵌入式关系型数据库引擎,它不需要一个独立的服务器进程,可以直接访问数据库文件。在 Python 中,我们可以使用内置的 sqlite3 模块来操作 SQLite 数据库。在本篇文章中,我们将介绍如何使用 Python 操作 SQLite 数据库,包括创建数据库、创建表、插入数据、查询数据等操作。

安装 sqlite3 模块

首先,我们需要安装 sqlite3 模块。在大多数情况下,sqlite3 模块已经包含在 Python 的标准库中,无需额外安装。如果你的 Python 环境中没有该模块,你可以通过以下命令安装:

pip install pysqlite3

连接到数据库

在操作 SQLite 数据库之前,我们首先需要连接到数据库。我们可以使用 sqlite3.connect() 方法来连接到指定的数据库文件。如果该文件不存在,将会自动创建一个新的数据库文件。以下是一个连接到数据库的示例代码:

import sqlite3

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

# 创建一个游标对象
cursor = conn.cursor()

# 关闭数据库连接
conn.close()

创建表

在数据库中创建表是存储数据的基础。我们可以使用 CREATE TABLE 语句来创建一个新的表。以下是一个创建表的示例代码:

# 创建表
cursor.execute('''
CREATE TABLE users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL,
    age INTEGER
)
''')

插入数据

一旦表已经创建,我们可以向表中插入数据。我们可以使用 INSERT INTO 语句来插入新的数据记录。以下是一个插入数据的示例代码:

# 插入数据
cursor.execute('''
INSERT INTO users (name, age) VALUES (?, ?)
''', ('Alice', 30))

# 提交更改
conn.commit()

查询数据

查询数据是操作数据库中最常用的操作之一。我们可以使用 SELECT 语句来查询数据记录。以下是一个查询数据的示例代码:

# 查询数据
cursor.execute('SELECT * FROM users')
for row in cursor.fetchall():
    print(row)

关闭数据库连接

在完成数据库操作后,我们需要关闭数据库连接以释放资源。我们可以使用 close() 方法来关闭数据库连接。以下是一个关闭数据库连接的示例代码:

# 关闭数据库连接
conn.close()

示例

下面是一个完整的示例代码,演示了如何使用 Python 操作 SQLite 数据库:

import sqlite3

# 连接到数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

# 创建表
cursor.execute('''
CREATE TABLE users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL,
    age INTEGER
)
''')

# 插入数据
cursor.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Alice', 30))
cursor.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Bob', 25))

# 查询数据
cursor.execute('SELECT * FROM users')
for row in cursor.fetchall():
    print(row)

# 关闭数据库连接
conn.close()

关系图

erDiagram
    users {
        INTEGER id
        TEXT name
        INTEGER age
    }

饼状图

pie
    title 数据分布
    "Alice" : 30
    "Bob" : 25

通过上面的示例代码和图表,我们可以看到如何使用 Python 操作 SQLite 数据库。SQLite 是一个简单易用的数据库引擎,适用于小型项目和快速原型开发。希望本文对你有所帮助,谢谢阅读!