csv파일을 DB에 넣기(pandas이용)

전체 코드

coninfo = get_info('secret.json')
cnx = create_engine(f"mysql+pymysql://{coninfo['MYSQL_USER']}:{coninfo['MYSQL_PASSWORD']}@{coninfo['MYSQL_HOST']}:{coninfo['MYSQL_PORT']}/{coninfo['MYSQL_DBNAME']}")
df = pd.read_csv('filename.csv', encoding='utf8')
print(df.head(5))
df.to_sql(name=f"mayo", con=cnx, if_exists="replace", index=False)

 

코드 설명

 

coninfo = get_info('secret.json')

secret.json파일이 외부에 있고 DB접속정보를 json파일형태로 저장해둔다.

cnx = create_engine(f"mysql+pymysql://{coninfo['MYSQL_USER']}:

{coninfo['MYSQL_PASSWORD']}@{coninfo['MYSQL_HOST']}:{coninfo['MYSQL_PORT']}/{coninfo['MYSQL_DBNAME']}")

sqlalchemy라이브러리를 활용하여 DB에 연결

 

 

df = pd.read_csv('filename.csv', encoding='utf8')

pandas내 read_csv함수를 이용하여 파일을 읽고 데이터프레임 형식으로 저장한다.

df.to_sql(name=f"mayo", con=cnx, if_exists="replace", index=False)

데이터프레임형식의 데이터를 DB에 넣는다.