백엔드 프레임워크 & 언어/Python

Python-Mysql과 연동하여 Pandas DataFrame으로 변환

anodos 2023. 7. 24. 19:11

import mysql.connector
import pandas as pd

# MySQL 서버에 연결
conn = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database'
)

# 커서 생성
cursor = conn.cursor()

# 숫자로만 이루어진 테이블명은 backtick 기호(`)로 감싸준다.
table_name = '2023'

# SELECT 쿼리 실행
query = f"SELECT * FROM `{table_name}`"
cursor.execute(query)

# 결과 가져오기
result = cursor.fetchall()

# 컬럼 이름 가져오기
columns = [desc[0] for desc in cursor.description]

# 결과를 Pandas DataFrame으로 변환
df = pd.DataFrame(result, columns=columns)

# 결과 출력
print(df)

# 커서와 연결 닫기
cursor.close()
conn.close()

반응형