在Jupyter Notebook中使用SQL,你可以通过以下步骤创建一个数据库并连接到该数据库:
1. 安装SQLite和SQLAlchemy
SQLite 是一个轻量级的数据库系统,适合本地小型项目。SQLAlchemy 是一个强大的数据库连接工具,可以在Jupyter中方便地运行SQL查询。
在Jupyter Notebook的单元格中运行以下命令来安装:
!pip install sqlalchemy sqlite
2. 创建SQLite数据库
SQLite数据库是一个文件,我选择直接在本地创建一个SQLite数据库文件。
import sqlite3
# 创建一个名为 example.db 的SQLite数据库
conn = sqlite3.connect('example.db')
# 创建游标对象,用于执行SQL语句
cursor = conn.cursor()
# 创建一个示例表格
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER NOT NULL
)
''')
# 插入一些数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 25)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 30)")
# 提交事务保存数据
conn.commit()
# 关闭连接
conn.close()
3. 连接到数据库并查询
再次打开连接并执行SQL查询。
# 重新连接到数据库
conn = sqlite3.connect('example.db')
# 使用 pandas 方便读取 SQL 查询结果
import pandas as pd
# 查询数据
query = "SELECT * FROM users"
df = pd.read_sql_query(query, conn)
# 展示数据
print(df)
# 关闭连接
conn.close()