[python] csv의 데이터를 DB insert

간단하게 csv에 있는 데이터를 DB에 insert를 해보겠습니다.

궁극적으로 제가 python을 공부하는 이유는 csv데이터를 자유자제로 구성하고 클리닝 및 정제하며 DB에 insert 및 update를 회사에서 해야하는데 일단 간단하게 해본 것입니다.

일단 소스부터 보겠습니다.

import scipy.io

import csv

import pymysql

conn = pymysql.connect(host='127.0.0.1', user='root', password='******',

                       db='test', charset='utf8')

curs = conn.cursor()

conn.commit()

f = open('insert.csv','r')

csvReader = csv.reader(f)

for row in csvReader:

    user_no = (row[0])

    user_pred = (row[1])

    print (user_pred)

    print (user_no)

    

    sql = """insert into mat (id, pred) values (%s, %s)"""

    curs.execute(sql, (user_no, user_pred))

#db의 변화 저장

conn.commit()

    

f.close()

conn.close()

1. csv,mysql 를 사용하기 위해 import

2. db 연결

3. csv 파일 open

4. csv 파일의 no, pred 값을 각각 user_no, user_pred 변수에 저장

5. db insert 쿼리 부분을 보면 excute 함수와 플레이스 홀더(%s)를 활용하여 삽입 명령문을 실제 db에 데이터를 삽입한다.

아래의 사진은 csv 데이터 및 db에 insert 된 모습이다.

csv 파일	mysql db

이로써 간단하게 csv를 읽어 db에 insert를 했습니다.

교수님이 pandas를 활용하여 데이터를 읽고 소스를 보니 pandas를 활용하는게 많아 다음 포스팅에는

pandas로 데이터를 다루는것을 포스팅 하겠습니다.