01 주피터 노트북에서 파이썬으로 sql 태스트 환경 만들기¶

01 주피터 노트북에서 파이썬으로 sql 태스트 환경 만들기¶

• sql 코딩테스트를 준비하기 위해 DB룰 설치하고 heidi SQL 등을 설치하고, DB를 만드는 과정이 번거롭다고 느끼면 아래와 같이 해보세요!
• 단, 파이썬 코드에 익숙해야 유리합니다.

In [1]:

# sqlite3 패키지를 활용하여 example.db 만들기
import sqlite3
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

• 위 표와 같은 데이터베이스 테이블을 생성해 보겠습니다

In [2]:

# 쿼리를 이용한 테이블 생성 및 값 입력
table_create = """
    CREATE TABLE name_table(
        id INTEGER PRIMATY KEY,
        name VARCHAR(50) NOT NULL
    );
    """
insert_value = """
    INSERT INTO name_table (id, name)
    VALUES
    (1, 'Phil'),
    (2, 'John'),
    (3, 'William')
    """
cursor.execute(table_create)
cursor.execute(insert_value)
# example.db에 name_table 테이블 정보 및 name_table의 값 커밋
conn.commit()

• select 문을 이용하여 DB 내 데이터 여부 확인

In [3]:

cursor.execute(
    # write sql query
    """
    SELECT *
    FROM name_table
    """
)

rows = cursor.fetchall()
rows

Out[3]:

[(1, 'Phil'), (2, 'John'), (3, 'William')]

• 판다스를 이용하여 sql 조회결과를 데이터프레임 형태로 확인하기

In [4]:

import pandas as pd
# 컬럼이름 리스트로 만들기
column_names = [desc[0] for desc in cursor.description]
column_names

Out[4]:

['id', 'name']

In [5]:

# 컬럼이름과 rows를 활용하여 데이터 프레임 형태로 조회 결과 표현하기
df = pd.DataFrame(rows, columns=column_names)
df

Out[5]:

	id	name
0	1	Phil
1	2	John
2	3	William