알고리즘 트레이딩 (파이썬)

14. pandas와 SQLite: pandas의 DataFrame 객체를 DB에 저장하는 방법 (+ DataFrame, pandas, to_sql 메소드 등)

오이띠 2021. 4. 23. 09:49
728x90

1. 기초: pandas의 DataFrame은 2차원 형태의 자료구조

 

2. pandas의 DataFrame 객체를 DB에 저장하는 방법

 

 

           

             > import pandas as pd

             > from pandas import Series, DataFrame

                  # 만약 해당 줄을 실행하지 않는다면 다음과 같이 NameError가 난다. 

 

             > df = DataFrame(raw_data)

                    # raw_data라는 딕셔너리를 앞에서 만들고, DataFrame 생성자를 호출했다. 

                    # 파이썬 딕셔너리를 만들고, DataFrame 생성자를 호출하면 DataFrame 객체를 쉽게 생성가능

 

              > import sqlite3

              > con = sqlite3.connect(r"C:\Users\LG\kospi.db")

                    # sqlite3 모듈 임포트 후, connect함수를 호출해 기존의 kospi.db를 연다.

 

              > df.to_sql('test', con)

                     # 구조: [데이터프레임이름].to_sql('[테이블이름]', [DB이름])

                     # 'kospi.db'의 'test'라는 테이블에 df 객체의 데이터를 저장

                     # to_sql 메서드의 파라미터는 다음 그림을 참고

to_sql 메서드의 파라미터

 

       + 참고로, DataFrame 에러 발생시, df = to_sql('test', con, chunksize = 1000) 으로 실행한다!!

 

 

 

 

 

 

 

3. DataFrame객체가 DB(kospi.db)에 잘 저장되었는지 확인

                   [DB Browser for SQLite]를 열어, test라는 테이블이 생성되었음을 알 수 있다. 

 

 

 

                        [데이터 보기]를 통해, 'test'라는 테이블의 데이터를 확인했다. 

728x90