시스템 개발 및 관리/프로젝트 구조 설정

Python enum – 상수를 깔끔하게 관리하는 방법

Data Jun 2025. 4. 29. 09:51

Python에서는 여러 고정된 값을 다룰 때 enum(열거형)을 사용할 수 있습니다.
이 글에서는 enum이 무엇인지, 왜 쓰는지 간단히 정리해봅니다.

 

enum이란?

**enum(enumeration)**은

"이름이 붙은 상수들의 집합" 을 만들어주는 기능입니다.

import enum

class Models(enum.Enum):
    XGBOOST = enum.auto()
    ENCODER = enum.auto()
    LOGISTIC = enum.auto()

 

  • Models.XGBOOST, Models.ENCODER처럼 의미 있는 이름으로 상수를 관리할 수 있어요.
  • enum.auto()를 사용하면 값은 자동으로 부여됩니다.

 

enum을 쓰는 이유

 

 

 

예시

def train_model(model_type):
    if model_type == Models.XGBOOST:
        print("Training XGBoost Model")
    elif model_type == Models.LOGISTIC:
        print("Training Logistic Regression")

 

  • Models.XGBOOST처럼 명확한 이름으로 비교하니까
  • 실수를 줄이고 코드가 읽기 쉬워집니다.

 

한 줄 요약

 

✔️ enum은 고정된 상수들을 안전하고 가독성 좋게 관리하기 위한 도구다.

 


마무리

  • 단순한 문자열 대신 enum을 쓰면 코드의 안정성, 가독성, 유지보수성이 모두 좋아집니다.
  • 실무에서도 모델 종류, 상태 코드, 옵션 값 관리 등에 매우 자주 사용됩니다. 🚀