[Python] ENUM 타입 사용하기

파이썬에서 Enum(열거형) 타입을 사용하려면 enum 모듈을 사용해야 한다.

enum 모듈은 파이썬 3.4부터 표준 라이브러리에 포함되어 있다.

아래 코드는 Enum 타입을 정의하고 사용하는 간단한 예제

from enum import Enum

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

print(Color.RED)  # Color.RED
print(Color.RED.value)  # 1
print(Color['GREEN'])  # Color.GREEN
print(list(Color))  # [<Color.RED: 1>, <Color.GREEN: 2>, <Color.BLUE: 3>]

 

위 예제에서 Color 클래스는 Enum을 상속하고 있다. Enum 클래스의 각 멤버는 Enum 상수를 나타내며, 상수의 이름과 값을 정의할 수 있다.

Color.RED와 같이 멤버에 접근하면 해당 Enum 멤버를 얻을 수 있다. value 속성을 사용하면 해당 멤버의 값을 얻을 수 있다.

Color['GREEN']과 같이 멤버 이름을 문자열로 사용하여 멤버에 접근할 수도 있다.

list(Color)는 Enum 클래스의 모든 멤버를 리스트로 반환한다.

이와 같이 Enum 타입을 사용하면 상수의 집합을 정의하고, 해당 상수들을 유연하게 사용할 수 있다.