Amazon RDS MySQL 데이터베이스에서 SQL 쿼리를 실행하고 결과를 Pandas DataFrame으로 변환하는 과정을 보여준다.

mysql_to_pandas.py

import pandas as pd
from sqlalchemy import create_engine

def query_to_dataframe(host, port, username, password, database, query):
    """
    주어진 SQL 쿼리를 실행하고 결과를 pandas DataFrame으로 반환한다.

    매개변수:
    host (str): 데이터베이스 서버 호스트 주소
    port (int): 데이터베이스 서버 포트 번호
    username (str): 데이터베이스 사용자 이름
    password (str): 데이터베이스 사용자 비밀번호
    database (str): 데이터베이스 이름
    query (str): 실행할 SQL 쿼리

    반환값:
    pandas.DataFrame: 쿼리 결과를 담은 DataFrame
    """
    # SQLAlchemy 엔진을 사용하여 데이터베이스에 연결
    engine = create_engine(f"mysql+pymysql://{username}:{password}@{host}:{port}/{database}")

    # SQL 쿼리 실행 및 결과 반환
    df = pd.read_sql(query, engine)
    return df

# 사용 예시
host = "your-rds-hostname"
port = 3306
username = "your-username"
password = "your-password"
database = "your-database"
query = "SELECT * FROM your_table"

df = query_to_dataframe(host, port, username, password, database, query)
print(df)