[Python] DB 테이블 데이터를 DataFrame으로 읽어오기

들어가기 앞서,

• pymysql은 Python에서 MySQL 데이터베이스와 연동하는 데 사용되는 라이브러리입니다.

• pymysql을 사용하기 위해선 먼저 pymysql 라이브러리를 설치해야 합니다.

    pip install pymysql

• 또한, 데이터베이스 쿼리 결과를 pandas의 DataFrame으로 다루기 위해선 pandas 라이브러리도 함께 설치해야 합니다.

    pip install pandas

예제 코드

• MySQL 서버에 연결하고 데이터베이스 쿼리 결과를 DataFrame으로 저장하는 예제 코드를 다뤄보겠습니다.

    import pymysql
    import pandas as pd
  
    # MySQL 서버에 연결
    conn = pymysql.connect(
        host='your_mysql_host',
        user='your_mysql_username',
        password='your_mysql_password',
        db='your_mysql_database',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor
    )
  
    try:
        # 커서 생성
        with conn.cursor() as cursor:
  
            # 레코드 조회 쿼리 실행
            select_query = 'SELECT * FROM `employees`;'
            cursor.execute(select_query)
            result = cursor.fetchall()
  
            # 조회 결과를 DataFrame으로 변환
            df = pd.DataFrame(result)
  
            # DataFrame 출력
            print(df)
  
    finally:
        # 연결 닫기
        conn.close()