Python機器學習（八十三）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