본문 바로가기

파이썬으로 퀀트 프로그램 만들기 project

pandas로 파이썬에 SQL사용하기 - 데이터 프레임을 SQL DB에 저장

728x90

데이터 프레임 형태의 데이터를 sql db에 저장하는 방법을 배워봅시다.

 

1. 예시 데이터 불러오기

저는 seaborn에서 제공하는 iris데이터 프레임을 가져와서 사용하겠습니다.

이것은 임의의 데이터 이므로 여러분은 원하시는 데이터 프레임을 사용하시면 되겠습니다.

import seaborn as sns 
iris = sns.load_dataset('iris')

 

 

2. create engine

# engine = create_engine('mysql+pymysql://[사용자명]:[비밀번호]@[호스트:포트]/[사용할 데이터베이스]')
engine = create_engine('mysql+pymysql://root:1234@127.0.0.1:3306/nameofdb')

 

 

3. 데이터를 sql db에 저장

iris.to_sql(name = 'iris', con = engine, index = False, if_exists = 'replace')

index = False 해줘야 자동으로 첫번째 열에 index열이 생기는 것을 방지할 수 있습니다.
if_exists = 'replace' 를 통해 이미 데이터가 있으면 덮어씌우는 것으로 설정합니다.

 

 

4. 결과 확인

sql db에 iris 데이터가 잘 저장되었는지 확인합니다.

결과 이미지

잘 저장 되어있는 것을 확인할 수 있습니다.