포스트

파이썬에서 enum 사용법 이해하기

enum 이란 무엇인가?

enum은 ‘열거형’이라는 의미로, 프로그래밍에서 여러 상수 값을 하나의 이름으로 묶는 방법입니다. 이렇게 하면 코드가 더 읽기 쉬워지고 유지 관리하기도 쉽습니다. 예를 들어, 빨강, 녹색, 파랑과 같은 색상을 나타내고 싶을 때 각 색상을 1, 2, 3 이라는 숫자로 표현하는 것보다는 RED, GREEN, BLUE라는 이름을 사용하는 것이 더 이해하기 쉽습니다.

파이썬에서 enum 사용하는 방법

파이썬에서는 enum 모듈을 통해 열거형을 쉽게 사용할 수 있습니다. 이 모듈을 사용하면 미리 정의된 상수 집합을 만들고, 이를 통해 코드를 더 명확하게 할 수 있습니다.

1
2
3
4
5
6
from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

위 코드에서 Color라는 이름의 열거형을 정의하고, 이 안에 RED, GREEN, BLUE라는 멤버를 가지고 있습니다. 각 멤버는 각각 1, 2, 3이라는 값을 가집니다.

enum 멤버와 값에 접근하는 방법

열거형을 정의한 후에는 다음과 같이 열거형의 멤버나 값에 접근할 수 있습니다.

1
2
3
4
5
# 멤버에 접근
my_color = Color.RED

# 멤버의 값에 접근
value = Color.RED.value  # 결과는 1

주의할 점

  1. enum 멤버는 유일해야 합니다. 같은 이름이나 값의 멤버를 중복해서 만들 수 없습니다.
  2. enum은 불변(immutable)입니다. 즉, 정의한 후에는 수정할 수 없습니다.

요약

enum은 코드의 가독성과 유지 보수성을 높이는 좋은 방법입니다. 파이썬에서는 enum 모듈을 통해 쉽게 열거형을 사용할 수 있습니다. 이를 활용하면 프로그램의 로직을 더 명확하게 표현할 수 있습니다.

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.