快捷導(dǎo)航

Python深入淺出分析enum枚舉類(lèi)

更新時(shí)間：2022年07月21日 09:20:56 作者：Flyme?awei

在python中枚舉是一種類(lèi)（Enum,IntEnum），存放在enum模塊中。枚舉類(lèi)型可以給一組標(biāo)簽賦予一組特定的值，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

枚舉類(lèi)Enum

枚舉類(lèi)，在企業(yè)開(kāi)發(fā)中用的比較多

當(dāng)我們需要定義常量時(shí)，一個(gè)辦法是用大寫(xiě)的變量通過(guò)整數(shù)來(lái)定義，例如月份：

# 當(dāng)項(xiàng)目中需要使用12個(gè)月份時(shí)
JAN = 1
FEB = 2
...
NOV = 11
DEC = 12

但是這樣的定義的類(lèi)型是 int ，并且仍然是變量，并且在運(yùn)算中，無(wú)法時(shí)時(shí)創(chuàng)建對(duì)應(yīng)的值，當(dāng)然這種指代是以更好的方式去使用變量數(shù)值。

這里存在更好的方法是為這樣的枚舉類(lèi)型定義一個(gè)class類(lèi)型，然后，每個(gè)常量都是class的一個(gè)唯一實(shí)例。Python提供了 Enum 類(lèi)來(lái)實(shí)現(xiàn)這個(gè)功能。

優(yōu)化如下

# -*- coding: utf-8 -*-
from enum import Enum
# 枚舉中，一個(gè)名字對(duì)應(yīng)一個(gè)值
Month = Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Otc', 'Nov', 'Dec'))

這樣我們就獲得了 Month 類(lèi)型的枚舉類(lèi)，可以直接使用 Month.Jan 來(lái)引用一個(gè)常量，或者枚舉它的所有成員通過(guò)for循環(huán)進(jìn)行取值，

Month.__members__是Month的內(nèi)置變量，可以打印成員標(biāo)簽，

如果在這個(gè)方法調(diào)用上.items()函數(shù)所有成員都打印出來(lái)，數(shù)據(jù)類(lèi)型為元組

for name, member in Month.__members__.items():

print(f'{name} => {member} , {member.value}')+

定義一個(gè)顏色常量枚舉

# -*- coding: utf-8 -*-
from enum import Enum
# 枚舉中 一個(gè)一個(gè)名字對(duì)應(yīng)一個(gè)值
Month = Enum('抬頭-Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))
print(Month.__members__)  # 枚舉中的值自動(dòng)從1開(kāi)始，不會(huì)重復(fù)
# 得到三月份的值
print(Month['Mar'].value)  # 3
# 根據(jù)(2)來(lái)獲得月份名字
print(Month(2).name)  # Feb
# 定義一個(gè)顏色的常量枚舉
class Color(Enum):  # 第二種:自定義一個(gè)枚舉類(lèi)
    red = 100
    green = 200
    blue = 300
    yellow = 200  # 不允許key相同或者value,如果value重復(fù)，根據(jù)value取name只能取第一個(gè)
print(Color(200))  # Color.green

不允許key相同或者value相同,如果value重復(fù)，根據(jù)value取name只能取第一個(gè)