[python] 파이썬에서 데이터베이스 연동 시 CSV 파일 데이터 다루기

12 Dec 2023

python

파이썬은 데이터베이스와의 연동을 위해 다양한 방법을 제공합니다. 데이터베이스 연동을 테스트하기 위해 CSV 파일 데이터를 활용하는 방법에 대해 살펴보겠습니다.

CSV 파일 읽기

파이썬에서 CSV 파일을 읽기 위해 csv 모듈을 활용할 수 있습니다. 아래는 CSV 파일을 읽고 데이터를 출력하는 간단한 예제 코드입니다.

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

데이터베이스에 데이터 입력

CSV 파일에서 읽은 데이터를 데이터베이스에 입력하기 위해 sqlite3 모듈을 활용할 수 있습니다. 아래의 예제는 CSV 파일을 읽고 SQLite 데이터베이스에 데이터를 입력하는 방법을 보여줍니다.

import csv
import sqlite3

conn = sqlite3.connect('example.db')
cur = conn.cursor()

with open('data.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        cur.execute("INSERT INTO table_name (col1, col2, col3) VALUES (?, ?, ?)", (row[0], row[1], row[2]))

conn.commit()
conn.close()

데이터베이스에서 데이터 조회

데이터베이스에 입력된 데이터를 조회하기 위해서는 다음과 같이 SQL 쿼리를 실행할 수 있습니다.

import sqlite3

conn = sqlite3.connect('example.db')
cur = conn.cursor()
cur.execute("SELECT * FROM table_name")
rows = cur.fetchall()
for row in rows:
    print(row)
conn.close()

이제 CSV 파일을 읽고 데이터베이스와 연동하여 데이터를 다루는 방법에 대해 알아보았습니다.

더 많은 정보는 csv 모듈 공식 문서와 sqlite3 모듈 공식 문서를 참고하시기 바랍니다.