Python枚举类Enum超详细入门与进阶全攻略

一、什么是枚举(Enum)?

Enum(枚举),是用于表示一组有名称的常量类型。每个枚举成员都由唯一的名字和值组成,Python从3.4版本开始加入了标准库enum模块。

枚举优点:

  • 可读性强,表达清晰
  • 防止魔法数字、常量重复
  • 程序更安全、更自解释
  • 可与类型提示、IDE轻松集成

二、Enum 最简用法入门

python 复制代码
from enum import Enum

class Weekday(Enum):
    MONDAY = 1
    TUESDAY = 2
    WEDNESDAY = 3

print(Weekday.MONDAY)        # Weekday.MONDAY
print(Weekday.MONDAY.value)  # 1
print(Weekday.MONDAY.name)   # 'MONDAY'

无需担心值重复或非法赋值!


三、高级用法:自动赋值、唯一性与自定义属性

1. 自动赋值

python 复制代码
from enum import auto, Enum

class Weekday(Enum):
    MONDAY = auto()
    TUESDAY = auto()
    WEDNESDAY = auto()

auto会自动赋值,省去手动编号烦恼!

2. 保证唯一性

python 复制代码
from enum import unique,Enum

@unique
class Status(Enum):
    NORMAL = 1
    ERROR = 2
    CANCEL = 3
    PENDING = 1  #duplicate values found in <enum 'Status'>: PENDING -> NORMAL

print(Status.NORMAL)
print(Status.NORMAL.value)
# 如果有重复值会抛出duplicate values

3. 添加方法和属性

你还可以给枚举添加方法或属性,实现更强大功能:

python 复制代码
class VIP(Enum):
    BRONZE = 1
    SILVER = 2
    GOLD = 3

    def level(self):
        return {1: '初级', 2: '中级', 3: '高级'}.get(self.value, '未知')

vip = VIP.GOLD
print(vip.level())   # '高级'

4. 枚举类的比较大小

  • 普通Enum里面的排序是无序的,不支持排序
  • IntEnum支持排序,可以比较大小
python 复制代码
from enum import Enum, IntEnum

class User(IntEnum): # IntEnum 如果改成Enum,下面的sorted就会发生异常
   a = 98
   b = 30
   c = 12

try:
   print('\n'.join('  ' + s.name for s in sorted(User)))
except TypeError as err:
   print(' Error : {}'.format(err))

四、枚举的多种类型小结

  • Enum:最基础的通用枚举
  • IntEnum:值必须是int,兼容大小比较等整型操作
  • Flag / IntFlag:位运算枚举,适合状态组合或权限位

IntEnum 示例

python 复制代码
from enum import IntEnum
class Level(IntEnum):
    LOW = 0
    MED = 1
    HIGH = 2

print(Level.LOW < Level.HIGH)  # True

Flag示例(位操作)

python 复制代码
from enum import Flag, auto
class Permission(Flag):
    READ = auto()
    WRITE = auto()
    EXECUTE = auto()

perm = Permission.READ | Permission.WRITE
print(perm)  # Permission.READ|WRITE

五、枚举的遍历与转换

遍历所有成员

python 复制代码
for day in Weekday:
    print(day, day.value)
# 输出 Weekday.MONDAY 1 ... 等

由值获取(反差)

python 复制代码
day = Weekday(1)    # 获取Weekday.MONDAY

如果你喜欢这篇文章,欢迎点赞、收藏和转发,更多Python干货内容敬请关注!

相关推荐
用户21411832636021 分钟前
01-开源版COZE-字节 Coze Studio 重磅开源!保姆级本地安装教程,手把手带你体验
前端
我要学习别拦我~9 分钟前
kaggle分析项目:steam付费游戏数据分析
python·游戏·数据分析
大模型真好玩15 分钟前
深入浅出LangChain AI Agent智能体开发教程(四)—LangChain记忆存储与多轮对话机器人搭建
前端·人工智能·python
love530love17 分钟前
命令行创建 UV 环境及本地化实战演示—— 基于《Python 多版本与开发环境治理架构设计》的最佳实践
开发语言·人工智能·windows·python·conda·uv
帅夫帅夫39 分钟前
深入理解 JWT:结构、原理与安全隐患全解析
前端
都叫我大帅哥40 分钟前
深度学习的"Hello World":多层感知机全解指南
python·深度学习
都叫我大帅哥1 小时前
LangChain分层记忆解决方案:完整案例
python·langchain
Struggler2811 小时前
google插件开发:如何开启特定标签页的sidePanel
前端
小王子10241 小时前
Django实时通信实战:WebSocket与ASGI全解析(下)
python·websocket·django
爱编程的喵1 小时前
深入理解JSX:从语法糖到React的魔法转换
前端·react.js