Python3 查询数据库获取某一列的集合
在数据科学和软件开发中，经常需要从数据库中查询数据。Python3 提供了多种方式来实现这一需求，其中使用 sqlite3 模块是一个简单且常用的方法。本文将介绍如何使用 Python3 的 sqlite3 模块查询数据库，并获取某一列的集合。
环境准备
首先，确保你的系统中安装了 Python3。可以通过在终端或命令提示符中输入以下命令来检查 Python3 是否已安装：
python3 --version
如果未安装，可以从 [Python 官网]( 下载并安装。
创建数据库和表
在开始查询之前，我们需要创建一个数据库和表。以下是一个简单的示例，创建一个名为 example.db 的 SQLite 数据库，并在其中创建一个名为 users 的表：
import sqlite3
# 连接到 SQLite 数据库
# 如果文件不存在，会自动在当前目录创建:
conn = sqlite3.connect('example.db')
# 创建一个 Cursor:
cursor = conn.cursor()
# 执行一条 SQL 命令，创建 users 表:
cursor.execute('CREATE TABLE users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)')
# 关闭 Cursor:
cursor.close()
# 提交事务:
conn.commit()
# 关闭 Connection:
conn.close()
插入数据
接下来，我们向 users 表中插入一些数据：
import sqlite3
# 连接到 SQLite 数据库:
conn = sqlite3.connect('example.db')
# 创建一个 Cursor:
cursor = conn.cursor()
# 插入数据:
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 25)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Charlie', 35)")
# 提交事务:
conn.commit()
# 关闭 Cursor:
cursor.close()
# 关闭 Connection:
conn.close()
查询某一列的集合
现在，我们可以使用 sqlite3 模块查询 users 表中的 name 列，并获取其集合：
import sqlite3
# 连接到 SQLite 数据库:
conn = sqlite3.connect('example.db')
# 创建一个 Cursor:
cursor = conn.cursor()
# 执行 SQL 查询:
cursor.execute("SELECT name FROM users")
# 获取查询结果:
names = cursor.fetchall()
# 打印结果:
print("Names:", names)
# 关闭 Cursor:
cursor.close()
# 关闭 Connection:
conn.close()
结果展示
执行上述代码后，你将看到类似以下输出：
Names: [('Alice',), ('Bob',), ('Charlie',)]
这是一个包含所有用户名称的元组列表。你可以根据需要对这些数据进行进一步的处理和分析。
总结
通过本文的介绍，你应该已经了解了如何使用 Python3 的 sqlite3 模块查询数据库，并获取某一列的集合。这种方法简单易学，适用于多种场景。当然，除了 SQLite，Python3 还支持其他类型的数据库，如 MySQL、PostgreSQL 等。你可以根据自己的需求选择合适的数据库和相应的 Python 库来实现数据查询和处理。
