一文帶你了解Python枚舉類enum的使用
更新時間：2022年10月26日 15:22:03 作者：XerCis
枚舉是與多個唯一常量綁定的一組符號，因為枚舉表示的是常量，建議枚舉成員名用大寫 IntEnum 便于進行系統(tǒng)交互。本文將通過一些示例代碼講解Python枚舉類的使用，需要的可以參考一下
簡介
枚舉是與多個唯一常量綁定的一組符號
因為枚舉表示的是常量，建議枚舉成員名用大寫
IntEnum 便于進行系統(tǒng)交互
初試
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
print(Color.RED) # Color.RED
print(repr(Color.RED)) # <Color.RED: 1>
print(type(Color.RED)) # <enum 'Color'>
print(isinstance(Color.GREEN, Color)) # True
print(Color.RED.name) # RED
遍歷
from enum import Enum
class Shake(Enum):
VANILLA = 7
CHOCOLATE = 4
COOKIES = 9
MINT = 3
for shake in Shake:
print(shake)
# Shake.VANILLA
# Shake.CHOCOLATE
# Shake.COOKIES
# Shake.MINT
__members__ 列出 name 和成員
from enum import Enum
class Shape(Enum):
SQUARE = 2
DIAMOND = 1
CIRCLE = 3
ALIAS_FOR_SQUARE = 2
for name, member in Shape.__members__.items():
print(name, member)
# SQUARE Shape.SQUARE
# DIAMOND Shape.DIAMOND
# CIRCLE Shape.CIRCLE
# ALIAS_FOR_SQUARE Shape.SQUARE
可哈希
枚舉成員可哈希，可用于字典和集合
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
apples = {}
apples[Color.RED] = 'red delicious'
apples[Color.GREEN] = 'granny smith'
print(apples)
# {<Color.RED: 1>: 'red delicious', <Color.GREEN: 2>: 'granny smith'}
訪問成員
- name：變量名
- value：值
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
# 通過值訪問
print(Color(1)) # Color.RED
print(Color(3)) # Color.BLUE
# 通過name訪問
print(Color['RED']) # Color.RED
print(Color['GREEN']) # Color.GREEN
# 訪問成員的name或value
print(Color.RED.name) # RED
print(Color.RED.value) # 1
唯一枚舉值
裝飾器 @unique
from enum import Enum, unique
@unique
class Mistake(Enum):
ONE = 1
TWO = 2
THREE = 3
FOUR = 3
# ValueError: duplicate values found in <enum 'Mistake'>: FOUR -> THREE
自動枚舉值
如果枚舉值不重要，可以用 auto()， 默認從 1 開始
from enum import Enum, auto
class Color(Enum):
RED = auto()
BLUE = auto()
GREEN = auto()
print(Color(1)) # Color.RED
print(list(Color)) # [<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 3>]
比較運算
- Enum 不支持直接排序
- IntEnum 可直接排序
Enum
from enum import Enum
class Color(Enum):
RED = 1
BLUE = 2
GREEN = 3
print(Color.RED is Color.RED) # True
print(Color.RED is Color.BLUE) # False
print(Color.RED is not Color.BLUE) # True
try:
Color.RED < Color.BLUE # Enum不支持直接排序
except Exception as e:
print(e) # '<' not supported between instances of 'Color' and 'Color'
print(Color.BLUE == Color.RED) # Fasle
print(Color.BLUE != Color.RED) # True
print(Color.BLUE == Color.BLUE) # True
print(Color.BLUE == 2) # False
IntEnum
from enum import IntEnum
class Color(IntEnum):
RED = 1
BLUE = 2
GREEN = 3
# IntEnum可直接排序
print(Color.RED < Color.BLUE) # True
print(Color.BLUE == 2) # True
功能性API
from enum import Enum
class Animal(Enum):
ANT = 1
BEE = 2
CAT = 3
DOG = 4
Animal = Enum('Animal', 'ANT BEE CAT DOG') # 同上
print(Animal) # <enum 'Animal'>
print(Animal.ANT) # <Animal.ANT: 1>
print(Animal.ANT.value) # 1
print(list(Animal)) # [<Animal.ANT: 1>, <Animal.BEE: 2>, <Animal.CAT: 3>, <Animal.DOG: 4>]
IntEnum
除了不可以直接和 Enum 比較，其他都類似整數(shù)
from enum import Enum, IntEnum
class Color(Enum):
RED = 1
GREEN = 2
class Shape(IntEnum):
CIRCLE = 1
SQUARE = 2
class Request(IntEnum):
POST = 1
GET = 2
print(Shape.CIRCLE == Color.RED) # False
print(Shape == 1) # False
print(Shape.CIRCLE == 1) # True
print(Shape.CIRCLE == Request.POST) # True
print(int(Shape.CIRCLE)) # 1
print(['a', 'b', 'c'][Shape.CIRCLE]) # b
print([i for i in range(Shape.SQUARE)]) # [0, 1]
IntFlag
類似 int，IntFlag 成員使用按位運算符得到的結果仍然是 IntFlag 成員
IntFlag 和 Enum 的一個區(qū)別在于，如果沒有設置任何標志（值為 0），則其布爾值為 False
from enum import IntFlag
class Perm(IntFlag):
R = 4 # 讀
W = 2 # 寫
X = 1 # 執(zhí)行
print(Perm.R | Perm.W) # <Perm.R|W: 6>
print(Perm.R + Perm.W) # 6
RW = Perm.R | Perm.W
print(Perm.R in RW) # True
from enum import IntFlag
class Perm(IntFlag):
R = 4 # 讀
W = 2 # 寫
X = 1 # 執(zhí)行
RWX = 7 # 讀寫執(zhí)行
print(Perm.RWX) # <Perm.RWX: 7>
print(~Perm.RWX) # <Perm.-8: -8>
print(Perm.R & Perm.X) # <Perm.0: 0>
print(bool(Perm.R & Perm.X)) # False
print(Perm.X | 8) # <Perm.8|X: 9>
Flag
- Flag 與 IntFlag 類似，成員可使用按位運算符進行組合，但不可與其他 Flag 或 int 組合
- 推薦用 auto() 作為值
from enum import Flag, auto
class Color(Flag):
BLACK = 0
RED = auto()
BLUE = auto()
GREEN = auto()
WHITE = RED | BLUE | GREEN
print(Color.RED & Color.GREEN) # <Color.0: 0>
print(bool(Color.RED & Color.GREEN)) # False
print(Color.WHITE) # <Color.WHITE: 7>
print(Color.BLACK) # <Color.BLACK: 0>
print(bool(Color.BLACK)) # False
知識點
不支持同名
以上就是一文帶你了解Python枚舉類enum的使用的詳細內(nèi)容，更多關于Python枚舉類enum的資料請關注腳本之家其它相關文章！
相關文章
這篇文章主要介紹了Python reversed函數(shù)及使用方法解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-03-03
這篇文章主要介紹了Python 字符串操作，所謂字符串，就是由0個或者多個字符組成的有限序列，字符串的字符可以是特殊符號、英文字母、中文字符、日文的平假名或片假名、希臘字母、Emoji字符等等。下面我們大家一起來學習文章詳細內(nèi)容吧2021-11-11
今天小編就為大家分享一篇python 切換root 執(zhí)行命令的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
為OpenCV是搞計算機視覺必須要掌握的基礎,這篇文章主要給大家介紹了關于OpenCV2學習筆記之視頻流讀取與處理的相關資料,文中通過示例代碼介紹的非常詳細,需要的朋友可以參考下2022-01-01
Python+flask編寫一個簡單實用的自動排班系統(tǒng)
這篇文章主要為大家詳細介紹了如何基于Python+flask編寫一個簡單實用的自動排班系統(tǒng),文中的示例代碼講解詳細,有需要的小伙伴可以了解下2025-03-03
這篇文章主要介紹了Python netmiko模塊的使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-02-02
django authenticate用戶身份認證的項目實踐
Django的contrib.auth模塊中的authenticate()函數(shù)用于對用戶的憑據(jù)進行身份驗證,本文就來介紹一下django authenticate用戶身份認證的使用,具有一定的參考價值,感興趣的可以了解一下2023-08-08
python完成FizzBuzzWhizz問題(拉勾網(wǎng)面試題)示例
這篇文章主要介紹了python完成FizzBuzzWhizz問題(拉勾網(wǎng)面試題)示例,需要的朋友可以參考下2014-05-05
函數(shù)cv::mean計算數(shù)組元素的平均值M,每個通道都是獨立的,并返回這個平均值,這篇文章主要給大家介紹了關于OpenCV計算平均值cv::mean的相關資料,需要的朋友可以參考下2021-08-08
這篇文章主要介紹了python?中的?super,提到 super，最直接的想法就是它代表了父類，替父類執(zhí)行某些方法，但是理解也僅止步于此，下面對 super 做進一步理解，需要的朋友可以參考下2022-08-08