개요

이 글에서는 파이썬에서 Psycopg2를 사용하여 데이터베이스 테이블의 데이터를 확인하고 필터링하는 방법을 알아보겠습니다. Psycopg2는 파이썬에서 PostgreSQL 데이터베이스와의 상호 작용을 도와주는 라이브러리입니다.

Psycopg2 설치

먼저, Psycopg2를 설치해야 합니다. 아래의 명령어를 사용하여 설치할 수 있습니다.

pip install psycopg2

데이터베이스 연결

데이터베이스에 연결하기 위해 psycopg2.connect() 함수를 사용합니다. 연결을 위해 데이터베이스의 호스트, 포트, 사용자 이름, 비밀번호, 데이터베이스 이름 등을 제공해야 합니다.

import psycopg2

# 데이터베이스 연결 정보
conn = psycopg2.connect(
    host="localhost",
    port="5432",
    user="username",
    password="password",
    database="database_name"
)

데이터 조회 및 필터링

psycopg2.connect() 함수를 사용하여 데이터베이스에 연결한 후, cursor() 메서드로 커서를 만들어 데이터베이스와 상호 작용합니다.

cur = conn.cursor()

전체 데이터 조회

테이블에서 전체 데이터를 조회하려면 execute() 메서드를 사용하여 SQL 쿼리를 실행합니다.

cur.execute("SELECT * FROM table_name")
rows = cur.fetchall()

for row in rows:
    print(row)

조건에 따른 데이터 필터링

특정 조건에 따라 데이터를 필터링하려면 SQL 쿼리에 조건을 추가합니다. 예를 들어, 나이가 30살 이상인 사용자의 데이터를 조회하는 방법은 다음과 같습니다.

cur.execute("SELECT * FROM table_name WHERE age >= 30")
filtered_rows = cur.fetchall()

for row in filtered_rows:
    print(row)

사용 후 연결 종료

데이터베이스 작업을 마치면 Psycopg2 커넥션을 닫아야 합니다.

cur.close()
conn.close()

결론

이렇게 파이썬에서 Psycopg2를 사용하여 데이터베이스 테이블의 데이터를 확인하고 필터링하는 방법을 알아보았습니다. 데이터베이스 연결, 데이터 조회, 데이터 필터링 등의 작업을 통해 데이터를 손쉽게 다룰 수 있습니다.