[python] PyMySQL을 이용하여 데이터베이스의 레코드를 특정 기준으로 집계하는 방법을 알아본다.

04 Dec 2023

python

데이터베이스는 정보를 구조화하여 저장하는 용도로 사용됩니다. 데이터베이스에서 저장된 데이터를 특정 기준으로 집계하는 과정은 매우 중요합니다. 이번에는 PyMySQL을 사용하여 데이터베이스의 레코드를 특정 기준으로 집계하는 방법에 대해 알아보도록 하겠습니다.

PyMySQL이란?

PyMySQL은 Python 프로그램에서 MySQL 데이터베이스에 접근하고 조작할 수 있도록 해주는 라이브러리입니다. 데이터베이스와 상호작용하기 위해 필요한 다양한 함수와 메서드를 제공하여 개발자가 간편하게 데이터베이스 작업을 수행할 수 있도록 도와줍니다.

레코드 집계하기

PyMySQL을 사용하여 데이터베이스의 레코드를 특정 기준으로 집계하는 방법은 크게 다음과 같은 단계로 이루어집니다:

데이터베이스에 연결하기
SQL 질의문 작성하기
집계 요청을 보내고 결과 받기
결과 처리하기

1. 데이터베이스에 연결하기

먼저, PyMySQL을 사용하여 데이터베이스에 연결해야 합니다. 연결하는 코드는 다음과 같습니다:

import pymysql

connection = pymysql.connect(
    host="localhost",
    user="your_username",
    password="your_password",
    database="your_database"
)

위의 코드에서 “localhost”, “your_username”, “your_password”, “your_database”는 각각 호스트, 사용자 이름, 비밀번호, 데이터베이스 이름에 해당하는 값을 입력해주어야 합니다. 이렇게 데이터베이스에 연결하면 다음 단계로 진행할 수 있습니다.

2. SQL 질의문 작성하기

다음으로는 집계를 수행하기 위한 SQL 질의문을 작성해야 합니다. 예를 들어, “orders”라는 테이블에서 “customer_id”를 기준으로 주문한 횟수를 집계하고 싶다면 다음과 같은 SQL 질의문을 작성할 수 있습니다:

query = "SELECT customer_id, COUNT(*) as order_count FROM orders GROUP BY customer_id"

위의 코드에서 “customer_id”, “orders”, “order_count”는 각각 테이블 이름, 기준이 되는 열 이름, 집계 결과로 사용할 열 이름입니다. 집계를 수행하고자 하는 테이블과 열의 이름에 맞게 SQL 질의문을 작성해야 합니다.

3. 집계 요청을 보내고 결과 받기

작성한 SQL 질의문을 PyMySQL을 통해 데이터베이스에 전송하여 집계 요청을 보내고 결과를 받아와야 합니다. 이를 위해 다음과 같은 코드를 사용할 수 있습니다:

with connection.cursor() as cursor:
    cursor.execute(query)
    result = cursor.fetchall()

위의 코드에서 “query”는 이전 단계에서 작성한 SQL 질의문입니다. “fetchall()” 메서드는 질의 결과를 모두 가져오는 역할을 합니다.

4. 결과 처리하기

마지막으로, 받아온 집계 결과를 처리하여 필요한 작업을 수행할 수 있습니다. 결과 처리는 사용자의 요구사항에 따라 다양하게 이루어질 수 있습니다. 예를 들어, 각 고객별로 주문한 횟수를 출력하는 코드는 다음과 같습니다:

for row in result:
    customer_id = row[0]
    order_count = row[1]
    print(f"Customer ID: {customer_id}, Order Count: {order_count}")

위의 코드는 “result”에 저장된 결과를 순회하며 각 고객 ID와 주문 횟수를 출력합니다. 필요한 작업에 맞게 결과를 처리하면 됩니다.

마무리

PyMySQL을 사용하여 데이터베이스의 레코드를 집계하는 방법을 알아보았습니다. 데이터베이스에 연결하고, SQL 질의문을 작성하여 집계를 요청하며, 결과를 받고 처리함으로써 원하는 결과를 얻을 수 있습니다. 데이터베이스 작업을 효율적으로 수행하기 위해 PyMySQL은 유용한 라이브러리입니다.