Pandas 读取 SQL 数据库

要从SQL数据库中加载数据，可以使用Pandas的read_sql_query方法。

我们将使用sqlite来测演示。

首先安装python的sqlite驱动 pysqlite3：

pip install pysqlite3

pysqlite3用于创建数据库连接，然后使用SELECT查询数据，加载DataFrame。

这里使用了database.db文件，要生成此文件，可以参考Pandas DataFrame存储到CSV, JSON，SQL。

import sqlite3

con = sqlite3.connect("database.db")

df = pd.read_sql_query("SELECT * FROM purchases", con)

df

输出

    index  apples  oranges
0   David       1        2
1    June       3        0
2    Lily       0        7
3  Robert       2        3

就像CSV一样，我们可以通过传递index_col='index'，来设置索引:

df = df.set_index('index')

df

输出

        apples  oranges
index
David        1        2
June         3        0
Lily         0        7
Robert       2        3