[python] PyMySQL을 이용하여 데이터베이스의 데이터를 그룹화하는 방법을 알아본다.

이번 글에서는 파이썬의 PyMySQL 라이브러리를 사용해서 데이터베이스의 데이터를 그룹화하는 방법에 대해 알아보겠습니다.

1. PyMySQL 설치

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

pip install PyMySQL

2. 데이터베이스 연결하기

PyMySQL을 사용하기 위해 우선 데이터베이스에 연결해야 합니다. 다음 예시 코드는 localhost에 위치한 데이터베이스에 연결하는 방법을 보여줍니다:

import pymysql

# 데이터베이스 연결
connection = pymysql.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database_name'
)

위의 코드 중 your_username, your_password, your_database_name에는 실제 데이터베이스의 접속 정보를 입력해야 합니다.

3. 데이터 그룹화

이제 데이터베이스의 데이터를 그룹화할 수 있습니다. GROUP BY 구문을 사용하여 원하는 열에 따라 데이터를 그룹화할 수 있습니다. 아래의 예시 코드는 users 테이블에서 age 열을 기준으로 데이터를 그룹화하는 방법을 보여줍니다:

try:
    # 커서 생성
    with connection.cursor() as cursor:
        # SQL 쿼리 실행
        sql = "SELECT age, COUNT(*) FROM users GROUP BY age"
        cursor.execute(sql)

        # 결과 가져오기
        result = cursor.fetchall()

        # 결과 출력
        for row in result:
            print(f"Age: {row[0]}, Count: {row[1]}")
finally:
    # 연결 종료
    connection.close()

위의 예시 코드에서는 users 테이블에서 age 열을 기준으로 데이터를 그룹화하고, 각 그룹의 레코드 수를 출력하고 있습니다.

4. 실행 결과

위의 예시 코드를 실행하면 다음과 같은 결과가 출력됩니다:

Age: 20, Count: 2
Age: 25, Count: 3
Age: 30, Count: 1

위의 결과는 users 테이블에서 age 열을 기준으로 그룹화한 결과입니다.

참고 자료