DB(Database)/PostgreSQL

[PostgreSQL] Python에서 PostgreSQL 접속하기, python에서 SQL 쿼리문 실행하기 + 쿼리 실행 결과 데이터프레임으로 만들기

탱젤 2021. 1. 13. 21:43
  • PostgreSQL 서버에 접속하기

psycopg2 설치

import psycopg2

conn_string = "host = 'localhost' dbname = 'db명' user = 'user명' password = '패스워드'"
conn = psycopg2.connect(conn_string)
cur = conn.cursor()
  • 위의 코드 실행 후 SQL 쿼리를 파이썬으로 실행 가능
import pandas as pd

cur.execute("쿼리문")
# 예시
# cur.execute("SELECT * FROM [DB명];")

result = cur.fetchall() # result에 결과 저장

# 쿼리문 실행 결과를 pandas dataframe 형식으로 만들기
my_df = pd.DataFrame(result) 
my_df.columns = [desc[0] for desc in cur.description]

아래는 내가 직접 실행해본 결과!

 

728x90