파이썬 mysql 연동 (pymysql, pandas)

mysql을 연동하고, 데이터프레임 형태로 보기 쉽게 노출하는 예제 코드입니다.

import pymysql
import pandas

db = pymysql.connect(
    user='root', 
    passwd='root', 
    host='127.0.0.1', 
    db='test_db', 
    charset = 'utf8'
)

cursor = db.cursor(pymysql.cursors.DictCursor)

sql = "select * from member"
cursor.execute(sql)

result = cursor.fetchall()
df = pandas.DataFrame(result)

# 전체 보기
print(df)

# 인덱스 보기
print(df.columns)

# 0~2열 보기
print(df.iloc[:, 0:3])

# 1, 4, 6행 보기
print(df.iloc[[1, 4, 6]])
print(df.iloc[[1, 4, 6], :])

# 0~2행 / 1, 4, 6열 보기
print(df.iloc[0:3, [1, 4, 6]])

# 인덱스 명으로 특정 열 보기
print(df.loc[:, ['id', 'name', 'birthday']])