Zenn
🙄

Pythonで列挙型（Enum）を使いこなす

2023/04/05に公開
Python
tech

はじめに：

プログラミングにおいて、固定された値のセットを表現する必要がしばしばあります。例えば、週の曜日や月、方角など。Pythonでは、これらの値を効率的に表現するために、「列挙型」（Enum）というデータ構造を使用します。本記事では、Pythonの列挙型の基本的な使い方を解説します。

1. 列挙型（Enum）とは？

列挙型（Enum）は、プログラムで一連の固定値を表現するための特殊なデータ型です。列挙型は、予め定義された値のセットを持ち、それ以外の値を受け入れません。これにより、コードの可読性が向上し、誤って無効な値を使用するリスクが減少します。

2. Pythonで列挙型を定義する方法

Pythonで列挙型を定義するには、enumモジュールを使用します。以下に、週の曜日を表す列挙型の例を示します。

from enum import Enum

class Weekday(Enum):
    MONDAY = 1
    TUESDAY = 2
    WEDNESDAY = 3
    THURSDAY = 4
    FRIDAY = 5
    SATURDAY = 6
    SUNDAY = 7

3. 列挙型の値にアクセスする

列挙型の値にアクセスするには、列挙型の名前と値の名前をドット（.）でつなげます。例えば、Weekday.MONDAYで、曜日の月曜日を表す値にアクセスできます。

today = Weekday.MONDAY
print(today)  # 出力: Weekday.MONDAY

4. 列挙型と制御構造

列挙型は、制御構造（if文やfor文など）と組み合わせて使用することができます。例えば、以下のようにif文で曜日を判断することができます。

today = Weekday.FRIDAY

if today == Weekday.SATURDAY or today == Weekday.SUNDAY:
    print("It's the weekend!")
else:
    print("It's a weekday.")

また、for文を使って列挙型のすべての値を繰り返し処理することもできます。以下に、Weekday列挙型のすべての値を表示する例を示します。

for day in Weekday:
    print(day)

このコードは、次のような出力を生成します。

Weekday.MONDAY
Weekday.TUESDAY
Weekday.WEDNESDAY
Weekday.THURSDAY
Weekday.FRIDAY
Weekday.SATURDAY
Weekday.SUNDAY

5. 列挙型の利用シーン

列挙型は、以下のようなシーンで特に役立ちます。

  1. 限定された選択肢を持つシチュエーション：
    列挙型は、選択肢が限られている場合に特に有用です。例えば、週の曜日、月、季節、方向（北、南、東、西）、またはプログラムで扱うカードのスート（ハート、ダイヤ、スペード、クローバー）などです。

  2. 状態管理：
    列挙型は、状態管理に役立ちます。例えば、オンラインショッピングアプリケーションで注文ステータスを追跡する場合、注文ステータスには「新規」「処理中」「発送済み」「キャンセル済み」などの固定値があります。列挙型を使用することで、状態を効果的に表現し、状態遷移を安全に管理できます。

  3. エラーコード：
    エラーコードを定義する際にも、列挙型が便利です。各エラーコードに固有の名前と数値を割り当てることで、エラー処理を容易に行うことができます。

  4. 設定オプション：
    アプリケーションの設定オプションを扱う場合、列挙型を使用することで、設定オプションの範囲を明確に制限し、不正な値を防ぐことができます。例えば、表示言語の選択肢やテーマカラーなどが挙げられます。

これらの固定値のセットが必要な場面で、列挙型を使用することで、コードの可読性と安全性が向上し、誤って無効な値を使用するリスクが減少します。

6. まとめ

本記事では、Pythonで列挙型の定義方法、アクセス方法、制御構造と組み合わせた使用方法を学びました。
列挙型（Enum）は、一連の固定値を表現するための効率的な方法を提供します。列挙型を使用することで、コードの可読性と安全性が向上し、誤って無効な値を使用するリスクが減少します。

Discussion