[Python][PostgreSQL] DB에서 데이터 Import

PostgreSQL에 연결해서 Dataframe으로 데이터 불러오기 

 

def query_postgreSQL(query):
    import pandas as pd
    import psycopg2 as pg
    from datetime import datetime

    # DB Connection
    conn = pg.connect(host='XXX.XXX.XX.XX',
                       port='XXXX',
                       dbname='dbname',
                       user='username',
                       password='pwpw')
                       
 #   start_tm = datetime.now()

     # Get a DataFrame
    query_result = pd.read_sql(query, conn)

#     end_tm = datetime.now()

#     print('START: ', str(start_tm))
#     print('END: ', str(end_tm))
#     print('ELAP: ', str(end_tm - start_tm))

    # Close connection
    conn.close()

    return query_result

tabname = 'sales'
field_list = ['product','price']

query = "SELECT " + field_list + " FROM " + tabname
    
## Excute SQL & Data Import in Python
df = query_postgreSQL(query)