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