pandas习题 056：将 sqlite 数据库数据读取到 DataFrame

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）以下代码为创建一个 sqlite 数据库（存储位置在内存），并创建一个 students 表写入一些数据，添加代码，将 students 表的数据构造到 pandas 的 DataFrame。

import sqlite3
import pandas as pd

# 使用上下文管理器连接到内存中的 SQLite 数据库
with sqlite3.connect(':memory:') as conn:
    # 创建一个表格
    conn.execute('''
    CREATE TABLE students (
        id INTEGER PRIMARY KEY,
        name TEXT,
        age INTEGER,
        grade TEXT
    )
    ''')

    # 插入假数据
    students_data = [
        (1, 'Alice', 20, 'A'),
        (2, 'Bob', 21, 'B'),
        (3, 'Charlie', 22, 'C'),
        (4, 'David', 23, 'B'),
        (5, 'Eva', 24, 'A')
    ]

    conn.executemany('INSERT INTO students VALUES (?, ?, ?, ?)', students_data)

参考答案

Python 代码如下：

import sqlite3
import pandas as pd

# 使用上下文管理器连接到内存中的 SQLite 数据库
with sqlite3.connect(':memory:') as conn:
    # 创建一个表格
    conn.execute('''
    CREATE TABLE students (
        id INTEGER PRIMARY KEY,
        name TEXT,
        age INTEGER,
        grade TEXT
    )
    ''')

    # 插入假数据
    students_data = [
        (1, 'Alice', 20, 'A'),
        (2, 'Bob', 21, 'B'),
        (3, 'Charlie', 22, 'C'),
        (4, 'David', 23, 'B'),
        (5, 'Eva', 24, 'A')
    ]

    conn.executemany('INSERT INTO students VALUES (?, ?, ?, ?)', students_data)

    # 查询数据并读取到 DataFrame
    # df = pd.read_sql_query('SELECT * FROM students', conn)
    df = pd.read_sql('SELECT * FROM students', conn)

# 显示 DataFrame
print(df)
'''
   id     name  age grade
0   1    Alice   20     A
1   2      Bob   21     B
2   3  Charlie   22     C
3   4    David   23     B
4   5      Eva   24     A
'''

df 就是 DataFrame 数据。

解析

查看相关链接中的知识。

（完）

pandas习题 056：将 sqlite 数据库数据读取到 DataFrame

参考答案

解析

相关内容