快捷導(dǎo)航

Python深入淺出分析enum枚舉類

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

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

枚舉類Enum

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

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

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

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

這里存在更好的方法是為這樣的枚舉類型定義一個(gè)class類型，然后，每個(gè)常量都是class的一個(gè)唯一實(shí)例。Python提供了 Enum 類來實(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 類型的枚舉類，可以直接使用 Month.Jan 來引用一個(gè)常量，或者枚舉它的所有成員通過for循環(huán)進(jìn)行取值，

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

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

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開始，不會(huì)重復(fù)
# 得到三月份的值
print(Month['Mar'].value)  # 3
# 根據(jù)(2)來獲得月份名字
print(Month(2).name)  # Feb
# 定義一個(gè)顏色的常量枚舉
class Color(Enum):  # 第二種:自定義一個(gè)枚舉類
    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è)