[MySQL / PostgreSQL] python & pandas와 연동하기

SQL을 python과 연동해서 작업하면 훨씬 수월하다. 

우선 MySQL을 python과 연동하는 법을 알아보자.

 

MySQL 라이브러리 설치

pip install pymysql

MySQL을 python과 연동하기

import pymysql

conn = pymysql.connect(host='host', user='user', port=port, password='password', db='db')

 

PostgreSQL 연동도 이와 비슷하다.

 

PostgreSQL 라이브러리 설치

pip install psycopg2

PostgreSQL을 python과 연동하기

import psycopg2 as pg

conn = pg.connect(host='host', user='user', port=port, password='password', dbname='dbname')

 

MySQL / PostgreSQL 저장된 데이터 테이블 pandas로 불러오기

sql = 'SELECT * FROM [table_name];'

#방법1
import pandas as pd
df = pd.read_sql_query(sql, conn)

#방법2
import pandas.io.sql as psql
df = psql.read_sql(sql, conn)