World's Best AI Learning Platform with profoundly Demanding Certification Programs
Designed by IITian's, only for AI Learners.
How to use Enum in python?
By Albert, a year ago
Bookmark
0
1 Answer
0
Neha
Install enum
pip install enum34
Creating an enum :
from enum import Enum
class Color(Enum):
red = 1
green = 2
blue = 3
print(Color.red) # Color.red
print(Color(1)) # Color.red
print(Color['red']) # Color.red
Enum Iteration :
Enums are iterable:
class Color(Enum):
red = 1
green = 2
blue = 3
[c for c in Color] # [<Color.red: 1>, <Color.green: 2>, <Color.blue: 3>]