mysql 테이블 판다스 데이터프레임으로 불러오기

mysql 테이블 데이터프레임으로 불러오기

sql 테이블을 데이터프레임으로 불러오는 방법입니다.
큰 틀로 본다면 mysql에 연결한 후 해당 서버에서 테이블을 pd.read_sql()을 통해서 가져옵니다.

pymysql을 이용한 방법

pymysql 라이브러리를 이용해 테이블을 불러오겠습니다.
pymysql로 mysql에 연결합니다.

import pandas as pd
import pymysql
# 데이터베이스 연결 설정

con = pymysql.connect(
    host='localhost',
    user='root',
    password='비밀번호',port= port, database='your_database'
)

# SQL 쿼리 실행 및 데이터프레임으로 변환 
query = "SELECT * FROM your_table" 
df = pd.read_sql(query, con)

con.close()

print(df)

con으로 pymysql 연결된 데이터베이스를 지정합니다. 테이블을 가져오는 쿼리를 작성한 후 read_sql을 통해 데이터프레임으로 불러옵니다.

sqlalchemy 를 이용한 방법

개인적으로 선호하는 방법입니다. sqalchemy를 이용하면 engine 객체를 따로 close하지 않아도 됩니다. pandas.read_sql() 함수는 engine 객체를 사용하여 내부적으로 연결을 열고 닫기 때문에, 사용자가 명시적으로 close()를 호출할 필요가 없습니다.

from sqlalchemy import create_engine
import pandas as pd

# MySQL 연결
engine = create_engine("mysql+mysqlconnector://your_username:your_password@localhost/your_database")

# SQL 쿼리 실행 및 데이터프레임으로 변환
quary = "SELECT * FROM your_table"
df = pd.read_sql(quary, engine)

# 데이터프레임 출력
print(df)

마치며

sql 연결 라이브러리만 있다면 read_sql로 손쉽게 데이터프레임으로 불러올 수 있습니다.

 

함께 보면 좋은 글

 

파이썬에서 Mysql 사용하는 간편한 방법(feat.pymysql)

 

SQL 기초 정리 DDL, DML, DCL 알아보기

 

빅데이터란? 확장되고 있는 정의