[Python] 파이썬에서 sql데이터 불러오기

파이썬에서 sql구문을 사용하여 데이터를 불러와야할 때가 있다.

 

 

sqlite3 패키지의 connect함수로 불러올 수 있다.

import sqlite3
import pandas as pd

sqlite 파일이 없으면 db파일에서 자동으로 생성해준다.

connection = sqlite3.connect("e-commerce.sqlite")
# connection = sqlite3.connect("e-commerce.db") -> 파일이 없으면 새로 만들게 됨.

query를 미리 작성 후 read_sql을 통해 위에서 불러온 connection과 연결시켜주면 데이터가 로드된다.

query = "SELECT * FROM 'orders'"

order = pd.read_sql(query, connection, index_col='id')

query가 길다면 """를 사용하여 입력해주면된다

query = """
    SELECT *
        FROM 'orders'
        WHERE price >= 1000
        OR state = 'confirmed'
        ORDER BY price DESC
        LIMIT 3
"""

order = pd.read_sql(query, connection, index_col='id')
order

 

이제 sql로 불러온 데이터를 활용하여 분석을 진행하면 된다.