본문 바로가기

파이썬/파이썬 예제

파이썬 코드를 활용한 DB 데이터를 CSV 파일로 저장

# 구성도

1. 파이썬 코드를 활용하여 특정 테이블 별 데이터를 추출한다.
- 아래 예제 코드에서 예를 들어보면 world.online_retail 테이블이 될 것이다.

2. 추출한 데이터를 CSV 파일로 저장한다.
- 아래 코드에서 예를 들어보면 write_online_retail_England.csv 파일과 write_online_retail_England.csv 파일이 될 것이다.

3. 저장된 파일을 특정 폴더로 이동한다.
- 아래 코드를 예를 들어보면 C:\Users\zzanggu\Desktop\나라별 저장\ 폴더에 저장이 된다.

# python pandas를 활용한 데이터 추출

import pandas as pd
import pymysql
from datetime import datetime

# 판다스 라이브러리를 활용한 데이터 추출 코드

# 시작시간 출력
print('start-time : ', str(datetime.now())[:19])

# DB 접속 정보
conn = pymysql.connect(host='127.0.0.1', user='root', password='1004rtrt$%', db='world', charset='utf8')

# United Kingdom 조건으로 조회되는 데이터를 write_online_retail_England.csv 파일로 저장
query = 'SELECT * FROM world.online_retail where country = "United Kingdom"'
df = pd.read_sql_query(query, conn)
df.to_csv(r'C:\Users\hist\Desktop\나라별 저장\write_online_retail_England.csv', index=False)

# France 조건으로 조회되는 데이터를 write_online_retail_France.csv 파일로 저장
query = 'SELECT * FROM world.online_retail where Country = "France"'
df = pd.read_sql_query(query, conn)
df.to_csv(r'C:\Users\zzanggu\Desktop\나라별 저장\write_online_retail_France.csv', index=False)

# 종료시간 출력
print('end-time : ', str(datetime.now())[:19])

- 간단한 코드를 활용하여 DB에 있는 데이터를 내려받고 CSV 파일로 저장하는 방법에 대해서 구현 하였다.

- 나아가 내려 받은 파일을 SFTP나 전송 프로그램을 이용하여 다른 경로로 이동하는 방법에 대해서도 구현을 한다면 자동으로 데이터 파일로 저장하고 활용하는 방법이 가능할거 같다.

- 끝 -