Python枚舉(enum)模塊使用的全面指南
更新時(shí)間：2025年07月30日 09:20:24 作者：Yant224
枚舉（Enumeration）是編程中一種重要的數(shù)據(jù)類(lèi)型,它允許開(kāi)發(fā)者定義一組命名的常量,Python 通過(guò) enum 模塊提供了強(qiáng)大的枚舉支持,本教程將全面講解 Python 枚舉的使用,從基礎(chǔ)到高級(jí)應(yīng)用,需要的朋友可以參考下
一、為什么需要枚舉？
1. 傳統(tǒng)常量的局限性
# 傳統(tǒng)方式定義常量
RED = 1
GREEN = 2
BLUE = 3
def print_color(color):
if color == RED:
print("紅色")
elif color == GREEN:
print("綠色")
elif color == BLUE:
print("藍(lán)色")
# 問(wèn)題：
print_color(1) # 正確
print_color(100) # 錯(cuò)誤但不會(huì)報(bào)錯(cuò)
2. 枚舉的優(yōu)勢(shì)
- 類(lèi)型安全：防止無(wú)效值
- 可讀性強(qiáng)：使用名稱(chēng)而非魔數(shù)
- 易于維護(hù)：集中管理相關(guān)常量
- IDE 支持：自動(dòng)補(bǔ)全和文檔提示
- 避免沖突：自動(dòng)處理值唯一性
二、基礎(chǔ)枚舉：Enum
1. 創(chuàng)建簡(jiǎn)單枚舉
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
# 使用枚舉
print(Color.RED) # Color.RED
print(Color.RED.name) # 'RED'
print(Color.RED.value) # 1
2. 枚舉的比較
# 身份比較 print(Color.RED is Color.RED) # True # 值比較 print(Color.RED == 1) # False print(Color.RED.value == 1) # True # 等值比較 print(Color.RED == Color.RED) # True print(Color.RED == Color.BLUE) # False
3. 遍歷枚舉
# 遍歷所有成員
for color in Color:
print(color.name, color.value)
# 輸出：
# RED 1
# GREEN 2
# BLUE 3
三、自動(dòng)賦值：auto()
auto() 自動(dòng)生成唯一值：
from enum import Enum, auto
class Direction(Enum):
NORTH = auto()
EAST = auto()
SOUTH = auto()
WEST = auto()
print(list(Direction))
# [<Direction.NORTH: 1>, <Direction.EAST: 2>, ...]
自定義自動(dòng)賦值
class CustomAuto(Enum):
def _generate_next_value_(name, start, count, last_values):
return f"{name}_{count}"
FIRST = auto() # 'FIRST_1'
SECOND = auto() # 'SECOND_2'
四、唯一枚舉：@unique
確保所有值唯一：
from enum import Enum, unique
@unique
class Status(Enum):
ACTIVE = 1
INACTIVE = 2
# PENDING = 1 # 會(huì)引發(fā) ValueError: duplicate values
五、功能枚舉：IntEnum 和 StrEnum
1. IntEnum - 整型枚舉
from enum import IntEnum
class HttpStatus(IntEnum):
OK = 200
NOT_FOUND = 404
SERVER_ERROR = 500
# 可以與整數(shù)比較
print(HttpStatus.OK == 200) # True
2. StrEnum (Python 3.11+)
from enum import StrEnum
class LogLevel(StrEnum):
DEBUG = 'DEBUG'
INFO = 'INFO'
WARNING = 'WARNING'
ERROR = 'ERROR'
# 可以與字符串比較
print(LogLevel.DEBUG == 'DEBUG') # True
六、標(biāo)志枚舉：Flag
用于位操作：
from enum import Flag, auto
class Permissions(Flag):
READ = auto() # 1
WRITE = auto() # 2
EXECUTE = auto() # 4
ALL = READ | WRITE | EXECUTE # 7
# 使用示例
user_perms = Permissions.READ | Permissions.WRITE
print(Permissions.READ in user_perms) # True
print(Permissions.EXECUTE in user_perms) # False
七、高級(jí)用法
1. 枚舉的別名
class Shape(Enum):
SQUARE = 1
RECTANGLE = 1 # 別名，與SQUARE相同
print(Shape.SQUARE is Shape.RECTANGLE) # True
2. 動(dòng)態(tài)創(chuàng)建枚舉
Animal = Enum('Animal', 'CAT DOG BIRD')
print(list(Animal))
# [<Animal.CAT: 1>, <Animal.DOG: 2>, <Animal.BIRD: 3>]
# 帶值創(chuàng)建
Vehicle = Enum('Vehicle', [('CAR', 10), ('BUS', 20), ('TRAIN', 30)])
3. 枚舉方法
class Planet(Enum):
MERCURY = (3.303e+23, 2.4397e6)
VENUS = (4.869e+24, 6.0518e6)
def __init__(self, mass, radius):
self.mass = mass
self.radius = radius
@property
def surface_gravity(self):
G = 6.67300E-11
return G * self.mass / (self.radius * self.radius)
print(Planet.MERCURY.surface_gravity)
4. 枚舉的序列化
import json
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
# 自定義JSON編碼
class EnumEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, Enum):
return obj.name
return super().default(obj)
# 序列化
print(json.dumps(Color.RED, cls=EnumEncoder)) # "RED"
# 反序列化
def enum_decoder(dct):
if 'color' in dct:
dct['color'] = Color[dct['color']]
return dct
data = '{"color": "RED"}'
obj = json.loads(data, object_hook=enum_decoder)
print(obj['color']) # Color.RED
八、實(shí)際應(yīng)用場(chǎng)景
1. 狀態(tài)管理
class OrderStatus(Enum):
PENDING = auto()
PROCESSING = auto()
SHIPPED = auto()
DELIVERED = auto()
CANCELLED = auto()
def update_order_status(order, new_status):
if not isinstance(new_status, OrderStatus):
raise ValueError("Invalid status")
# 更新邏輯...
2. 配置選項(xiàng)
class LogLevel(Enum):
DEBUG = 10
INFO = 20
WARNING = 30
ERROR = 40
def configure_logger(level=LogLevel.INFO):
# 配置日志...
3. Django 模型選擇字段
from django.db import models
from enum import Enum
class UserType(Enum):
REGULAR = 'R'
ADMIN = 'A'
GUEST = 'G'
@classmethod
def choices(cls):
return [(member.value, member.name) for member in cls]
class User(models.Model):
user_type = models.CharField(
max_length=1,
choices=UserType.choices(),
default=UserType.REGULAR.value
)
4. API 參數(shù)驗(yàn)證
from enum import Enum
from flask import Flask, abort
app = Flask(__name__)
class SortOrder(Enum):
ASC = 'asc'
DESC = 'desc'
@app.route('/api/users')
def get_users():
order = request.args.get('order', SortOrder.ASC.value)
try:
# 驗(yàn)證參數(shù)
sort_order = SortOrder(order)
except ValueError:
abort(400, "Invalid sort order")
# 處理請(qǐng)求...
九、枚舉與替代方案對(duì)比
| 特性 | 枚舉 | 字典 | 類(lèi)常量 |
|---|---|---|---|
| 類(lèi)型安全 | ? | ? | ? |
| 值唯一性 | ? | ? | ? |
| 可迭代性 | ? | ? | ? |
| 成員方法 | ? | ? | ? |
| IDE 支持 | ? | ? | ? |
| 序列化 | 需自定義 | 原生 | 原生 |
| 內(nèi)存效率 | 高 | 中 | 高 |
十、最佳實(shí)踐與常見(jiàn)陷阱
最佳實(shí)踐：
- 優(yōu)先使用
Enum替代魔數(shù) - 使用
@unique確保值唯一 - 為枚舉成員添加文檔字符串
- 使用
auto()簡(jiǎn)化值分配 - 在類(lèi)型提示中使用枚舉
常見(jiàn)陷阱：
值比較而非成員比較
# 錯(cuò)誤 if color == 1: ... # 正確 if color == Color.RED: ...
動(dòng)態(tài)修改枚舉
Color.RED = 10 # 引發(fā) AttributeError
忽略大小寫(xiě)
# 錯(cuò)誤
color = Color['red'] # KeyError
# 解決方法
class CaseInsensitiveEnum(Enum):
@classmethod
def _missing_(cls, value):
for member in cls:
if member.name.lower() == value.lower():
return member
return None
class Color(CaseInsensitiveEnum):
RED = 1
GREEN = 2
十一、性能考慮
枚舉在創(chuàng)建時(shí)有些開(kāi)銷(xiāo)，但使用時(shí)的性能與普通類(lèi)相當(dāng)：
# 創(chuàng)建性能
%timeit Enum('Color', 'RED GREEN BLUE')
# 10000 loops, best of 5: 29.7 μs per loop
# 訪問(wèn)性能
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
%timeit Color.RED
# 10000000 loops, best of 5: 29.9 ns per loop
十二、總結(jié)
Python 的 enum 模塊提供了強(qiáng)大而靈活的枚舉功能：
- 基礎(chǔ)枚舉：
Enum創(chuàng)建類(lèi)型安全的常量集合 - 自動(dòng)賦值：
auto()簡(jiǎn)化枚舉創(chuàng)建 - 特殊枚舉：
IntEnum,StrEnum,Flag滿足特定需求 - 高級(jí)特性：枚舉方法、自定義行為、動(dòng)態(tài)創(chuàng)建
- 應(yīng)用場(chǎng)景：狀態(tài)管理、配置選項(xiàng)、API設(shè)計(jì)等
使用決策樹(shù)
通過(guò)合理使用枚舉，您可以：
- 提高代碼可讀性和可維護(hù)性
- 減少錯(cuò)誤并增強(qiáng)類(lèi)型安全
- 創(chuàng)建更專(zhuān)業(yè)的API和接口
- 簡(jiǎn)化狀態(tài)管理和配置處理
枚舉是編寫(xiě)清晰、健壯代碼的重要工具。在您的下一個(gè)項(xiàng)目中嘗試使用枚舉，體驗(yàn)它帶來(lái)的代碼質(zhì)量提升！
以上就是Python枚舉(enum)模塊使用的全面指南的詳細(xì)內(nèi)容，更多關(guān)于Python枚舉(enum)使用的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
Python Scrapy實(shí)戰(zhàn)之古詩(shī)文網(wǎng)的爬取
本文將利用Python中Scrapy框架，實(shí)現(xiàn)爬取古詩(shī)文網(wǎng)上的詩(shī)詞數(shù)據(jù)，具體包括詩(shī)詞的標(biāo)題信息。文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下2022-05-05
Python借助AI快速開(kāi)發(fā)圖片轉(zhuǎn)PDF的工具
隨著數(shù)字文檔處理需求的激增,圖片轉(zhuǎn)PDF的需求日益廣泛,這篇文章主要為大家詳細(xì)介紹了Python如何借助工具快速開(kāi)發(fā)圖片轉(zhuǎn)PDF功能,感興趣的小伙伴可以了解下2025-08-08
使用Python實(shí)現(xiàn)一鍵往Word文檔的表格中填寫(xiě)數(shù)據(jù)
在工作中,我們經(jīng)常遇到將Excel表中的部分信息填寫(xiě)到Word文檔的對(duì)應(yīng)表格中,以生成報(bào)告,方便打印,所以本文小編就給大家介紹了如何使用Python實(shí)現(xiàn)一鍵往Word文檔的表格中填寫(xiě)數(shù)據(jù),文中有詳細(xì)的代碼示例供大家參考,需要的朋友可以參考下2023-12-12
Centos5.x下升級(jí)python到python2.7版本教程
這篇文章主要介紹了Centos5.x下升級(jí)python到python2.7版本教程,本文使用編譯安裝方式,并配置了一系列需要更改的配置項(xiàng),需要的朋友可以參考下2015-02-02
使用Python實(shí)現(xiàn)帶進(jìn)度條的異步多文件下載器
本文詳細(xì)介紹了如何使用?Python?+?asyncio?+?aiohttp?+?tqdm?實(shí)現(xiàn)一個(gè)?帶進(jìn)度條的異步多文件下載器,文中的示例代碼講解詳細(xì),有需要的可以了解下2025-11-11
Python?Playwright進(jìn)行常見(jiàn)的頁(yè)面交互操作
在使用?Playwright?進(jìn)行?Web?自動(dòng)化時(shí),頁(yè)面交互是核心操作之一,本文將詳細(xì)介紹如何使用?Playwright?進(jìn)行常見(jiàn)的頁(yè)面交互操作,希望對(duì)大家有所幫助2024-10-10![解決Python保存文件名太長(zhǎng)OSError: [Errno 36] File name too lon](http://img.jbzj.com/images/xgimg/bcimg6.png)
解決Python保存文件名太長(zhǎng)OSError: [Errno 36] File
這篇文章主要介紹了解決Python保存文件名太長(zhǎng)OSError: [Errno 36] File name too lon問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-05-05
Python利用Tiler制作專(zhuān)屬卡通頭像和LOGO
Tiler是一種使用各種其他較小圖像平鋪創(chuàng)建新圖像的工具，它與其他馬賽克工具不同，因?yàn)樗梢赃m應(yīng)多種形狀、大小、方向的貼圖，稱(chēng)為buil in build。本文就來(lái)利用Tiler制作專(zhuān)屬卡通頭像和LOGO，需要的可以參考一下2022-12-12
pytorch中使用cuda擴(kuò)展的實(shí)現(xiàn)示例
這篇文章主要介紹了pytorch中使用cuda擴(kuò)展的實(shí)現(xiàn)示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02
python讀取excel指定列數(shù)據(jù)并寫(xiě)入到新的excel方法
今天小編就為大家分享一篇python讀取excel指定列數(shù)據(jù)并寫(xiě)入到新的excel方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07