파이썬 SQLite3 DBMS로 저장하기

안녕하세요, 츄르 사려고 코딩하는 집사! 코집사입니다.

---

파이썬 SQLite3 DBMS로 저장하기

---

1. SQLite3 DBMS

> 파일시스템 기반의 경량 관계형 DBMS

> 경량 관계형 DBMS로 스마트폰 등의 embedded 환경에서 널리 사용

> 파이썬에서는 SQLite3가 기본 모듈로 제공

> 대용량 데이터 및 트랜잭션 처리 부적합

> 경량 데이터 및 트랜잭션 처리, 교육용 목적 등으로 사용

 

 

import pandas as pd
import sqlite3
from pandas.io import sql
import os

DB_NAME = 'top_cities.db'
TABLE_NAME = 'TOP_CITIES'

def db_save(df, db_name, table_name) :
    with sqlite3.connect(db_name) as con :
        try :
            df.to_sql(name = table_name, con = con, index = False, if_exists='append')
            #if_exists : {'fail', 'replace', 'append'} default : fail
        
        except Exception as e :
            print(str(e))
        print(len(df), '건 저장완료...')
        
def db_select(db_name, table_name):
    with sqlite3.connect(db_name) as con: 
        try:
            query = 'SELECT * FROM {}'.format(table_name)
            df = pd.read_sql(query, con = con)
        except Exception as e:
            print(str(e)) 
        return df  
        
def db_delete(db_name, table_name):
    with sqlite3.connect(db_name) as con: 
        try:
            cur = con.cursor()
            sql = 'DELETE FROM {}'.format(table_name)
            cur.execute(sql)
        except Exception as e:
            print(str(e)) 
top_cites = pd.read_csv('top_cities.csv')
db_save(top_cites, DB_NAME, TABLE_NAME)

df = db_select(DB_NAME, TABLE_NAME)
df