Python——枚举（enum）

最新推荐文章于 2025-09-12 00:22:01 发布

转载最新推荐文章于 2025-09-12 00:22:01 发布 · 932 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：https://www.cnblogs.com/-beyond/p/9777329.html

文章标签：

#python 枚举

Python 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了Python中如何使用枚举（Enum）类来提高代码的严谨性和安全性。枚举用于定义不可变的常量集合，避免了直接使用类变量可能导致的问题，如值的重复和外部修改。通过示例展示了枚举类的创建、访问和比较，强调了使用内置`enum`模块的重要性，以及其特性，如不允许枚举值的外部修改、枚举项的比较等。还提到了`IntEnum`和`unique`装饰器的用法，以及枚举类的成员遍历。最后，讨论了枚举在代码组织和可读性上的优势。

使用普通类直接实现枚举

　　在Python中，枚举和我们在对象中定义的类变量时一样的，每一个类变量就是一个枚举项，访问枚举项的方式为：类名加上类变量，像下面这样：

class color():
    YELLOW  = 1
    RED     = 2
    GREEN   = 3
    PINK    = 4
 
# 访问枚举项
print(color.YELLOW) # 1

虽然这样是可以解决问题的，但是并不严谨，也不怎么安全，比如：

　　1、枚举类中，不应该存在key相同的枚举项（类变量）

　　2、不允许在类外直接修改枚举项的值

class color():
    YELLOW  = 1
    YELLOW  = 3   # 注意这里又将YELLOW赋值为3，会覆盖前面的1
    RED     = 2
    GREEN   = 3
    PINK    = 4
 
# 访问枚举项
print(color.YELLOW) # 3
 
# 但是可以在外部修改定义的枚举项的值，这是不应该发生的
color.YELLOW = 99
print(color.YELLOW) # 99

解决方案：使用enum模块

　　enum模块是系统内置模块，可以直接使用import导入，但是在导入的时候，不建议使用import enum将enum模块中的所有数据都导入，一般使用的最多的就是enum模块中的Enum、IntEnum、unique这几项

# 导入枚举类
from enum import Enum
 
# 继承枚举类
class color(Enum):
    YELLOW  = 1
    BEOWN   = 1 
    # 注意BROWN的值和YELLOW的值相同，这是允许的，此时的BROWN相当于YELLOW的别名
    RED     = 2
    GREEN   = 3
    PINK    = 4
 
class color2(Enum):
    YELLOW  = 1
    RED     = 2
    GREEN   = 3
    PINK    = 4

使用自己定义的枚举类：

print(color.YELLOW) # color.YELLOW
print(type(color.YELLOW)) # <enum 'color'>
 
print(color.YELLOW.value)  # 1
print(type(color.YELLOW.value)) # <class 'int'>
 
print(color.YELLOW == 1)    # False
print(color.YELLOW.value == 1)  # True
print(color.YELLOW == color.YELLOW)  # True
print(color.YELLOW == color2.YELLOW)  # False
print(color.YELLOW is color2.YELLOW)  # False
print(color.YELLOW is color.YELLOW)  # True
 
print(color(1))         # color.YELLOW
print(type(color(1)))   # <enum 'color'>

注意事项如下：

　　1、枚举类不能用来实例化对象

　　2、访问枚举类中的某一项，直接使用类名访问加上要访问的项即可，比如 color.YELLOW

　　3、枚举类里面定义的Key = Value，在类外部不能修改Value值，也就是说下面这个做法是错误的

color.YELLOW = 2  # Wrong, can't reassign member

4、枚举项可以用来比较，使用==，或者is

　　5、导入Enum之后，一个枚举类中的Key和Value，Key不能相同，Value可以相，但是Value相同的各项Key都会当做别名，

　　6、如果要枚举类中的Value只能是整型数字，那么，可以导入IntEnum，然后继承IntEnum即可，注意，此时，如果value为字符串的数字，也不会报错：

from enum import IntEnum

7、如果要枚举类中的key也不能相同，那么在导入Enum的同时，需要导入unique函数

from enum import Enum, unique

8、打印枚举类所有值

for v in VIP.__members__:
    print(v)

如果输出所有值

for v in VIP.__members__.items():
    print(v)