要从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