How do I define an enumeration in Python?

How do I define an enumeration in Python?

You can define an enumeration in Python using the Enum class from the enum module. Here's a simple example:

from enum import Enum

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

This creates an enumeration called Color with three members.

Can you explain how to use the enumeration after defining it?

Sure! After defining the enumeration, you can use it like this:

print(Color.RED)  # Output: Color.RED
print(Color.RED.name)  # Output: 'RED'
print(Color.RED.value)  # Output: 1

You can also compare enumeration members.

What if I want to iterate over the enumeration members?

You can iterate over the members of an enumeration using a simple for loop:

for color in Color:
    print(color)

This will print each member of the Color enumeration.

Is it possible to add methods to an enumeration?

Yes, you can add methods to an enumeration. Here's an example:

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

    def describe(self):
        return f'This color is {self.name.lower()}.'

You can then call Color.RED.describe() to get a description.